Cod sursa(job #384255)

Utilizator cristikIvan Cristian cristik Data 19 ianuarie 2010 19:44:24
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <stdio.h>
#define max 15010

int arb[4*max+100],i,j,n,m,x,val,pos;
void update(int nod,int st,int dr)
{
    if(st==dr) { arb[nod]+=val; return; }
    int mid=(st+dr)>>1;
    if(pos<=mid) update(nod<<1,st,mid);
    if(pos>mid)  update((nod<<1)|1,mid+1,dr);

    arb[nod]=arb[nod<<1]+arb[(nod<<1)|1];
}
int query(int nod,int st,int dr)
{
    int val=0;
    if(i<=st && dr<=j) return arb[nod];
    int mid=(st+dr)>>1;
    if(i<=mid) val+=query(nod<<1,st,mid);
    if(j>mid)  val+=query((nod<<1)|1,mid+1,dr);

    return val;
}
int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(i=1; i<=n; i++)
    {
        pos=i; scanf("%d",&val);
        update(1,1,n);
    }
    for(; m>0; m--)
    {
        scanf("%d",&x);
        if(x)
        {
            scanf("%d%d",&i,&j);
            printf("%d\n",query(1,1,n));
        }
        else
        {
            scanf("%d%d",&pos,&val);
            val=-val;
            update(1,1,n);
        }
    }
    return 0;
}