Cod sursa(job #2632080)

Utilizator stefanvoicaVoica Stefan stefanvoica Data 2 iulie 2020 11:06:35
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.27 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin ("datorii.in");
ofstream fout("datorii.out");

struct abint
{
    int v[400002];

    void update (int nod,int st,int dr,int poz,int val)
    {
        if (st==dr)
        {
            v[nod]+=val;
            return ;
        }
        int mij=(st+dr)>>1;
        if (poz<=mij)
            update(2*nod,st,mij,poz,val);
        else
            update(2*nod+1,mij+1,dr,poz,val);
        v[nod]=v[2*nod]+v[2*nod+1];
    }

    int  query (int nod,int st,int dr,int a,int b)
    {
        if (b<st || dr< a)
            return 0;
        if (a<=st && dr<= b)
            return v[nod];
        int mij=(st+dr)>>1;
        if (b<=mij)
            return query(2*nod,st,mij,a,b);
        else if (a>mij)
            return query (2*nod+1,mij+1,dr,a,b);
        else
            return query(2*nod,st,mij,a,b)+query (2*nod+1,mij+1,dr,a,b);
    }

};
abint Arb;

int main()
{
    int n,i,x,q,t,a,b;
    fin>>n>>q;
    for (i=1; i<n+1; i++)
    {
        fin>>x;
        Arb.update(1,1,n,i,x);
    }
    for (; q>0; q--)
    {
        fin>>t>>a>>b;
        if (t==1)
            fout<<Arb.query(1,1,n,a,b)<<'\n';
        else
            Arb.update(1,1,n,a,-b);
    }
    return 0;
}