Cod sursa(job #564861)

Utilizator miticaMitica mitica Data 27 martie 2011 12:03:02
Problema Datorii Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.24 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)/2;
    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)/2;
    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;
}