Cod sursa(job #2751362)

Utilizator linte_robertLinte Robert linte_robert Data 14 mai 2021 20:38:36
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.26 kb

#include <iostream>
#include <fstream>
using namespace std;


std::ifstream f("exemplu.in");
std::ofstream g("zeap.out");

int t[60100];

void update(int nod, int st, int dr, int poz, int val) {
  if(st==dr) { t[nod]=val; return; }
  int mij = (st+dr)/2;
  if(poz<=mij) update(2*nod,st,mij,poz,val);
  else update(2*nod+1,mij+1,dr,poz,val);
  t[nod] = t[2*nod]+t[2*nod+1];
}

void update2(int nod, int st, int dr, int poz, int val)
{
    if(st==dr)
    {
        t[nod]-=val;
        return;
    }
    int mij = (st+dr)/2;
    if(poz<=mij) update2(2*nod,st,mij,poz,val);
    else update2(2*nod+1,mij+1,dr,poz,val);
    t[nod] = t[2*nod] + t[2*nod+1];
}
int getmax(int nod, int st, int dr, int l, int r) {
    if(l<=st && dr<=r) return t[nod];
    if( dr < l || st > r ) return 0;
    int v1 = 0, v2 = 0;
    int mij = (st+dr)/2;
    if(r>mij) v1 = getmax(2*nod,st,mij,l,r);
    if(l<=mij) v2 = getmax(2*nod+1,mij+1,dr,l,r);
    return v1+v2;
}



int main()
{
    int x, n, m, a, b, op;
    f >> n >> m;
    for(int i=1;i<=n;i++){
        f >> x;
        update(1,1,n,i+1,x);
    }
    for(int i = 1; i <= m; i++){
        f >> op >> a >> b;
        if( op == 0) update2(1,1,n,a,b);
        else g << getmax(1,1,n,a,b) << "\n";
    }
}