Cod sursa(job #1553801)

Utilizator daneel95Holteiu Daniel-Ninel daneel95 Data 20 decembrie 2015 15:52:41
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.35 kb
#include <fstream>
#define tata(x) (x)/2
#define fst(x) (x)<<1
#define fdr(x) ((x)<<1)+1

using namespace std;

ifstream in("datorii.in");
ofstream out("datorii.out");

int n,m,a[15005],arbint[4*15005];

void create_arbint(int poz,int st,int dr)
{
    if(st==dr)
    {
        arbint[poz]=a[st];
        return;
    }
    int mij=(st+dr)/2;
    create_arbint(fst(poz),st,mij);
    create_arbint(fdr(poz),mij+1,dr);
    arbint[poz]=arbint[fst(poz)]+arbint[fdr(poz)];
}

void update(int poz,int st,int dr,int vpoz,int val)
{
    if(st==dr)
    {
        arbint[poz]-=val;
        return;
    }
    int mij=(st+dr)/2;
    if(vpoz>mij) update(fdr(poz),mij+1,dr,vpoz,val);
        else update(fst(poz),st,mij,vpoz,val);
    arbint[poz]=arbint[fst(poz)]+arbint[fdr(poz)];
}

int query(int poz,int st,int dr,int a,int b)
{
    if(st>=a && dr<=b) return arbint[poz];
    int mij=(st+dr)/2;
    int sum=0;
    if(b>mij)
        sum = query(fdr(poz),mij+1,dr,a,b);
    if(a<=mij)
        sum+=query(fst(poz),st,mij,a,b);
    return sum;
}

int main()
{
    int i,q,x,y;
    in>>n>>m;
    for(i=1;i<=n;i++) in>>a[i];
    create_arbint(1,1,n);
    for(i=1;i<=m;i++)
    {
        in>>q>>x>>y;
        if(q==0) update(1,1,n,x,y);
        if(q==1) out<<query(1,1,n,x,y)<<"\n";
    }
    in.close();
    out.close();
    return 0;
}