Cod sursa(job #1425663)

Utilizator GinguIonutGinguIonut GinguIonut Data 27 aprilie 2015 21:00:55
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.09 kb
#include <fstream>
#define dim 15001
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int arb[4*dim],i,val,n,m,sum,poz,cond;
void upDate(int nod, int st, int dr)
{
    if(st==dr)
    {
        arb[nod]+=val;
        return;
    }
    int mijl=(st+dr)/2;
    if(poz<=mijl)
        upDate(2*nod,st,mijl);
    if(poz>mijl)
        upDate(2*nod+1,mijl+1,dr);
    arb[nod]=arb[nod*2]+arb[nod*2+1];
}
void query(int nod, int st, int dr)
{
    if(st>=poz&&dr<=val)
    {
        sum+=arb[nod];
        return;
    }
    int mijl=(st+dr)/2;
    if(poz<=mijl)
        query(2*nod,st,mijl);
    if(val>mijl)
        query(2*nod+1,mijl+1,dr);
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>val;
        poz=i;
        upDate(1,1,n);
    }
    for(i=1;i<=m;i++)
    {
        fin>>cond>>poz>>val;
        if(cond==0)
        {
            val*=-1;
            upDate(1,1,n);
        }
        else
        {
            sum=0;
            query(1,1,n);
            fout<<sum<<'\n';
        }
    }
    return 0;
}