Cod sursa(job #2132864)

Utilizator RaduGiucleaGiuclea Radu RaduGiuclea Data 16 februarie 2018 10:25:10
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <cstdio>
using namespace std;
int arb[50002],v[15002],sum;
inline void recalculate(int node)
{
    arb[node]=arb[2*node]+arb[2*node+1];
}
void build(int node,int st,int dr)
{
    if(st==dr)
        arb[node]=v[st];
    else
    {
        int mid=(st+dr)/2;
        build(2*node,st,mid);
        build(2*node+1,mid+1,dr);
        recalculate(node);
    }
}
void update(int poz,int k,int node,int st,int dr)
{
    if(st==dr)
        arb[node]-=k;
    else
    {
        int mid=(st+dr)/2;
        if(poz<=mid)
            update(poz,k,2*node,st,mid);
        else update(poz,k,2*node+1,mid+1,dr);
        recalculate(node);
    }
}
void query(int a,int b,int node,int st,int dr)
{
    if(a<=st&&dr<=b)
        sum+=arb[node];
    else
    {
        int mid=(st+dr)/2;
        if(a<=mid)
            query(a,b,node*2,st,mid);
        if(b>=mid+1)
            query(a,b,node*2+1,mid+1,dr);
    }
}
int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    int n,m,i,p,a,b;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
        scanf("%d",&v[i]);
    build(1,1,n);
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d",&p,&a,&b);
        if(!p)
            update(a,b,1,1,n);
        else
        {
            sum=0;
            query(a,b,1,1,n);
            printf("%d\n",sum);
        }
    }
    return 0;
}