Cod sursa(job #851404)

Utilizator andumMorie Daniel Alexandru andum Data 9 ianuarie 2013 22:45:17
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.25 kb
#include <cstdio>
#include <algorithm>
 
using namespace std;
 
int arb[400005],n,m,i,x,a,b,poz,val,s,f,cost;
 
void update(int nod, int st, int dr)
{
    int mij;
    if (st==dr)
    {
        arb[nod]-=val;
        return;
    }
    mij=(st+dr)>>1;
    if (poz<=mij)
        update(2*nod, st, mij);
    else
        update(2*nod+1, mij+1, dr);
    arb[nod]=arb[2*nod]+arb[2*nod+1];
}
 
void query(int nod, int st, int dr)
{
    int mij;
    if (s<=st && dr<=f)
    {
        cost+=arb[nod];
        return;
    }
    mij=(st+dr)>>1;
    if (s<=mij)
        query(2*nod, st, mij);
    if (mij<f)
        query(2*nod+1, mij+1, dr);
}
 
int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    scanf("%d %d", &n, &m);
    for (i=1; i<=n; ++i)
    {
        scanf("%d", &x);
        poz=i;
        val=-x;
        update(1,1,n);
    }
    for (i=1; i<=m; ++i)
    {
        scanf("%d %d %d", &x, &a, &b);
        if (x)
        {
            s=a;
            f=b;
            cost=0;
            query(1,1,n);
            printf("%d\n", cost);
        }
        else
        {
            poz=a;
            val=b;
            update(1,1,n);
        }
    }
 
    return 0;
}