Cod sursa(job #1815260)

Utilizator dianamichesaRosu Diana Michesa dianamichesa Data 24 noiembrie 2016 23:09:05
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.36 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int aint[60001],n,m,v[15001],pp;
int p(int n){
int k=1;
while(k<=n) k*=2;
return k;}
void build(int nod,int st,int dr)
{
    if(st==dr){
        aint[nod]=v[st];
        return;
    }
    else{
        int m=(st+dr)/2;
        build(nod*2,st,m);
        build(nod*2+1,m+1,dr);
        aint[nod]=aint[nod*2]+aint[nod*2+1];
        }
        return;
}
void update(int nod,int st,int dr,int poz,int val)
{
    if(st==dr){
        aint[nod]-=val;
        return;
    }
    else{
        int m=(st+dr)/2;
        if(poz<=m){
            update(2*nod,st,m,poz,val);
        }
        else{
            update(2*nod+1,m+1,dr,poz,val);
        }
        aint[nod]=aint[2*nod]+aint[2*nod+1];

    }
}
int query(int nod,int st,int dr,int p1,int p2)
{
    if(st>=p1 && dr<=p2){pp=aint[nod];return pp;}
    else{int m=(st+dr)/2,x=0,y=0;
    if(m>=p1) {x=query(2*nod,st,m,p1,p2);}
    if(m+1<=p2) {y=query(2*nod+1,m+1,dr,p1,p2);}
    return x+y;
    }
}

int main()
{
    f>>n>>m;
    for(int i=1;i<=n;i++) f>>v[i];
    build(1,1,n);
    for(int i=1;i<=m;i++){
        int k,x,y;
        f>>k>>x>>y;
        if(k==0) update(1,1,n,x,y);
        if(k==1){
            g<<query(1,1,n,x,y)<<'\n';
        }
    }
    return 0;
}