Cod sursa(job #2477314)

Utilizator Andy_ANDYSlatinaru Andrei Alexandru Andy_ANDY Data 19 octombrie 2019 22:48:16
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <bits/stdc++.h>
#define ll long long
using namespace std;
ifstream f ("datorii.in"); ofstream g ("datorii.out");
int n,m,arb[4*15000+5],v[15000+5],ansquery;
void build (int nod , int st,int dr)
{   if(st==dr)
    {   arb[nod]=v[st];
        return ;
    }
    int mij=(st+dr)>>1;
    build(2*nod,st,mij);
    build(2*nod+1,mij+1,dr);
    arb[nod]=arb[nod*2]+arb[nod*2+1];
    return ;
}
void query(int nod,int st,int dr,int l,int r)
{   if(l<=st and dr<=r)
    {   ansquery+=arb[nod];
        return ;
    }
    int mij=(st+dr)>>1;
    if(l<=mij) query(2*nod,st,mij,l,r);
    if(mij<r) query(2*nod+1,mij+1,dr,l,r);
    return ;
}
void update(int nod ,int st,int dr,int poz ,int val)
{   if(st==dr)
    {   arb[nod]-=val;
        return ;
    }
    int mij=(st+dr)>>1;
    if(mij>=poz) update(nod*2,st,mij,poz,val);
    else update(nod*2+1,mij+1,dr,poz,val);
    arb[nod]=arb[nod*2]+arb[nod*2+1];
    return ;
}
int main()
{   f>>n>>m;
    for(int i=1;i<=n;i++) f>>v[i];
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {   int c;
        f>>c;
        if(c==0)
        {   int poz,val;
            f>>poz>>val;
            update(1,1,n,poz,val);
        }
        else
        {   int st,dr;
            f>>st>>dr;
            ansquery=0;
            query(1,1,n,st,dr);
            g<<ansquery<<"\n";
        }
    }
    return 0;
}