Cod sursa(job #429711)

Utilizator AndrewTheGreatAndrei Alexandrescu AndrewTheGreat Data 30 martie 2010 13:29:40
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <iostream>
#define IN "datorii.in"
#define OUT "datorii.out"
#define Lg 15010

using namespace std;

int H[4*Lg];
int semn, Poz, val;
int N, M, Sum, S, D;

void update(int nod, int st, int dr)
{
    int mij;
    if(st==dr)
        H[nod]+=val*semn;
    else
    {
        mij=(st+dr)>>1;
        if(Poz<=mij)
            update(nod<<1,st,mij);
        else
            update((nod<<1)+1,mij+1,dr);
        H[nod]=H[nod<<1]+H[(nod<<1)+1];
    }
}

void citire()
{
    scanf("%d %d",&N,&M);
    semn=1;
    for(Poz=1;Poz<=N;Poz++)
    {
        scanf("%d",&val);
        if(val)
            update(1,1,N);
    }
}

void suma(int nod,int st,int dr)
{
    int mij;
    if(st>=S && dr<=D)
        Sum+=H[nod];
    else
    {
        mij=(st+dr)>>1;
        if(mij>=S)
            suma(nod<<1,st,mij);
        if(mij<D)
            suma((nod<<1)+1,mij+1,dr);
    }
}


void solve()
{
    semn=-1;
    int op;
    for(int i=1;i<=M;i++)
    {
        scanf("%d %d %d",&op,&S,&D);
        if(op)
        {
            Sum=0;
            suma(1,1,N);
            printf("%d\n",Sum);
        }
        else
        {
            Poz=S;
            val=D;
            update(1,1,N);
        }
    }
}

int main()
{
    freopen (IN,"r",stdin);
    freopen (OUT,"w",stdout);

    citire();
    solve();

    return 0;
}