Cod sursa(job #1703293)

Utilizator rotti321Rotar Mircea rotti321 Data 16 mai 2016 18:32:34
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,poz,val;
int TR[4*NMax];

void update(int nod,int st,int dr){
    if(poz <= st && dr <= poz){
        TR[nod] += val;
    }else{
        int mij = (st + dr) / 2;
        if(poz <= mij){
            update(2 * nod, st, mij);
        }
        if(poz > mij){
            update(2 * nod + 1, mij + 1, dr);
        }
        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 >> val;
        poz=i;
        update(1,1,n);
    }
    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;
            poz=x;
            update(1,1,n);
        }
    }
    return 0;
}