Cod sursa(job #1696911)

Utilizator VladTiberiuMihailescu Vlad Tiberiu VladTiberiu Data 30 aprilie 2016 12:25:56
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.25 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()
{
    f >> n >> m;
    for(int i = 1; i <= n; ++i){
        f >> a[i];
        update(1,1,n,i,i,a[i]);
    }
    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{
            a[x] -= y;
            update(1,1,n,x,x,a[x]);
        }
    }
    return 0;
}