Cod sursa(job #1017114)

Utilizator cat_red20Vasile Ioana cat_red20 Data 27 octombrie 2013 11:33:58
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.28 kb
#include<fstream>
#define MAXN 1<<15
using namespace std;
int s[MAXN],sol,x,y,n,m,p,val;
ifstream fin("datorii.in");
ofstream fout("datorii.out");

void update(int nod,int st,int dr)
{
    if(st==dr && st==p)
    {
        s[nod]+=val;
        return;
    }
    int m=(st+dr)/2;
    if(st<=p && p<=m)
    {
        update(2*nod,st,m);
        s[nod]+=val;
    }
    else
    if(m+1<=p && p<=dr)
    {
        update(2*nod+1,m+1,dr);
        s[nod]+=val;
    }
}

void citire()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>val;
        p=i;
        update(1,1,n);
    }
}

void querry(int nod,int st,int dr)
{
    if(x<=st && dr<=y)
    {
        //[st,dr] este inclus in [x,y]
        sol+=s[nod];
        return;
    }
    int m=(st+dr)/2;
    if(x<=m)
    querry(2*nod,st,m);
    if(y>m)
    querry(2*nod+1,m+1,dr);
}

void querries()
{
    int op,a,b;
    for(int i=1;i<=m;i++)
    {
        fin>>op>>a>>b;
        if(op==0)
        {
            val=-b;
            p=a;
            update(1,1,n);
        }
        else
        {
            sol=0;
            x=a; y=b;
            querry(1,1,n);
            fout<<sol<<"\n";
        }
    }
}

int main()
{
    citire();
    querries();
    return 0;
}