Cod sursa(job #1703287)

Utilizator rotti321Rotar Mircea rotti321 Data 16 mai 2016 18:28:24
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.26 kb
#include <bits/stdc++.h>

#define NMax 15002
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");

int n,m,p,x,y,sol;
int a[NMax],TR[4*NMax];

void update(int nod,int st,int dr,int a,int b,int v){
    if(a <= st && dr <= b){
        TR[nod] += v;
    }else{
        int mij = (st + dr) / 2;
        if(a <= mij){
            update(2 * nod, st, mij, a, b, v);
        }
        if(b > mij){
            update(2 * nod + 1, mij + 1, dr, a, b,v);
        }
        TR[nod] = TR[2 * nod] + TR[2 * nod + 1];
    }
}
void inter(int nod,int st,int dr,int a,int b){
    if(a <= st && dr <= b){
        sol += TR[nod];
    }else{
        int mij = (st + dr) / 2;
        if(a <= mij){
            inter(2 * nod, st, mij, a, b);
        }
        if(b > mij){
            inter(2 * nod + 1, mij + 1, dr, a, b);
        }
    }
}
int main()
{
    int val;
    f >> n >> m;
    for(int i = 1; i <= n; ++i){
        f >> val;
        update(1,1,n,i,i,val);
    }
    for(int i = 1; i <= m; ++i){
        f >> p >> x >> y;
        if(p == 1){
            sol = 0;
            inter(1,1,n,x,y);
            g << sol << '\n';
        }else{
            val = -y;
            update(1,1,n,x,x,val);
        }
    }
    return 0;
}