Cod sursa(job #904647)

Utilizator lianaliana tucar liana Data 4 martie 2013 17:53:47
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.23 kb
#include<stdio.h>
#define nmax 15005
#define pmax 33000
long i, n, m, a, b, val, poz, x, rez, op;
long s[pmax];

void update(long st, long dr, long nod)
{
    long mjc;
    if (st==dr)
        s[nod]+=val;
    else
    {
        mjc=(st+dr)/2;
        if (poz<=mjc)
            update(st,mjc,2*nod);
        if (mjc+1<=poz)
            update(mjc+1,dr,2*nod+1);
        s[nod]=s[2*nod]+s[2*nod+1];
    }
}

void query(long st, long dr, long nod)
{
    long mjc;
    if ((a<=st)&&(dr<=b))
        rez+=s[nod];
    else
    {
        mjc=(st+dr)/2;
        if (a<=mjc)
            query(st,mjc,2*nod);
        if (mjc+1<=b)
            query(mjc+1,dr,2*nod+1);
    }
}

int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    scanf("%ld %ld",&n,&m);
    for (i=1;i<=n;i++)
    {
        scanf("%ld",&x);
        poz=i;  val=x;
        update(1,n,1);
    }
    for (i=1;i<=m;i++)
    {
        scanf("%ld %ld %ld",&op,&a,&b);
        if (op==0)
        {
            poz=a;  val=-b;
            update(1,n,1);
        }
        if (op==1)
        {
            rez=0;
            query(1,n,1);
            printf("%ld\n",rez);
        }
    }
    return 0;
}