Cod sursa(job #2900780)

Utilizator Andoss1710Balanica Andrei Andoss1710 Data 12 mai 2022 09:38:39
Problema Datorii Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.06 kb
#include <iostream>
#include <fstream>

using namespace std;

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

long long int Segtree[100001], N, j, v[15001];

void adaugare(long long int nod, long long int poz, long long int left, long long int right){
    //cout<<left<<" "<<right<<endl;
    if(left > poz || right < poz)
        return;
    if(left == right){
        //cout<<nod<<endl;
        Segtree[nod] += v[poz];
        return;
    }
    adaugare(nod*2, poz, left, (left+right)/2);
    adaugare(nod*2+1, poz, (left+right)/2+1, right);
    Segtree[nod] = Segtree[2*nod] + Segtree[2*nod+1];
}
void constr(){
    for(j = 1; j <= N; j++){
        adaugare(1,j,1, N);
    }

}
void modif(long long int nod, long long int left, long long int right,long long int poz, long long int x){
    if(poz < left || poz > right){
            return;
    }
    if(right == left){
        Segtree[nod] -= x;
        return;
    }
    modif(nod*2, left, (right+left)/2,poz, x);
    modif(nod*2+1, (right+left)/2+1, right,poz, x);

    Segtree[nod] = Segtree[2*nod] + Segtree[2*nod+1];
}

int suma(long long int nod, long long int left, long long int right, long long int s, long long int r){
    if(s > right || left > r)
            //cout<<"acum"<<endl;
            return 0;
    if(s<=left && r>=right)
        return Segtree[nod];
    return suma(nod*2, left, (right+left)/2, s , r) + suma(nod*2+1, (right+left)/2+1, right, s, r);

}
int main()
{
    long long int M, poz, val, start, stop;
    fin>>N>>M;
    for(long long int i = 1; i<=N; i++)
        fin>>v[i];
    constr();
     //for(int i = 1; i<2*N; i++)
        //cout<<Segtree[i]<<" ";
    //cout<<endl;
    for(long long int i = 0; i<M; i++){
        long long int op;
        fin>>op;
        if(op==0){
            fin>>poz>>val;
            modif(1, 1, N, poz, val);
        }
        else{
            fin>>start>>stop;
            fout<<suma(1, 1, N, start, stop)<<"\n";
        }
        }
//    for(int i = 1; i<2*N; i++)
//        cout<<Segtree[i]<<" ";

    }