Cod sursa(job #2099021)

Utilizator ScarymovieMocanu Alexandru Scarymovie Data 3 ianuarie 2018 21:01:04
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.11 kb
#include <fstream>
#define NMAX 15000
using namespace std;
int v[4*NMAX+5];
void update(int nod,int st,int dr,int poz,int val)
{
    if(st+1==dr)
        {
            if(st==poz) v[nod]-=val;
            return ;
        }
    if(poz>=st && poz<dr)
        {
            int mij=st+(dr-st)/2;
            v[nod]-=val;
            update(2*nod,st,mij,poz,val);
            update(2*nod+1,mij,dr,poz,val);
        }
}
int query(int nod,int st,int dr,int left,int right)
{
    if(st>=dr) return 0;
    if(left<=st && right>=dr) return v[nod];
    if(dr<=left || right<=st) return 0;
    int mij=st+(dr-st)/2;
    int ans1=query(2*nod,st,mij,left,right);
    int ans2=query(2*nod+1,mij,dr,left,right);
    return ans1+ans2;
}
int main()
{
    int n,m,x,t,a,b;
    ifstream f("datorii.in");
    ofstream g("datorii.out");
    f>>n>>m;
    for(int i=1;i<=n;++i)
    {
        f>>x;
        update(1,1,n+1,i,-x);
    }
    for(int i=1;i<=m;++i)
        {
            f>>t>>a>>b;
            if(!t) update(1,1,n+1,a,b);
            else g<<query(1,1,n+1,a,b+1)<<'\n';
        }

    return 0;
}