Cod sursa(job #2771116)

Utilizator Raresr14Rosca Rares Raresr14 Data 25 august 2021 14:44:24
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.17 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int n,i,sol,v[15010],A[60010],op,q,a,b;
void build(int nod, int st, int dr){
    if(st==dr)
        A[nod]=v[st];
    else{
        int mid=(st+dr)/2;
        build(nod*2,st,mid);
        build(nod*2+1,mid+1,dr);
        A[nod]=A[nod*2]+A[nod*2+1];
    }
}
void update(int nod, int st, int dr){
    if(st==dr)
        A[nod]-=b;
    else{
        int mid=(st+dr)/2;
        if(a<=mid)
            update(nod*2,st,mid);
        if(a>mid)
            update(nod*2+1,mid+1,dr);
        A[nod]=A[nod*2]+A[nod*2+1];
    }
}
void query(int nod, int st, int dr){
    if(st>=a&&dr<=b)
        sol+=A[nod];
    else{
        int mid=(st+dr)/2;
        if(a<=mid)
            query(nod*2,st,mid);
        if(b>mid)
            query(nod*2+1,mid+1,dr);
    }
}
int main(){
    fin>>n>>q;
    for(i=1;i<=n;i++)
        fin>>v[i];
    build(1,1,n);
    for(;q--;){
        fin>>op>>a>>b;
        if(!op)
            update(1,1,n);
        else{
            sol=0;
            query(1,1,n);
            fout<<sol<<"\n";
        }
    }
    return 0;
}