Cod sursa(job #2901185)

Utilizator DafinaTrufasTrufas Dafina DafinaTrufas Data 13 mai 2022 10:59:52
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.21 kb
#include <iostream>
#include <fstream>
using namespace std;

int a[15000], d[60000];

void creare(int nod, int st, int dr)
{
    if(st == dr)
    {
        d[nod] = a[st];
        return;
    }
    creare(nod * 2, st, (st + dr) / 2);
    creare(nod * 2 + 1, (st + dr) / 2 + 1, dr);
    d[nod] = d[nod * 2] + d[nod * 2 + 1];
}

void achita(int nod, int st, int dr, int i, int val)
{
    if(i < st || i > dr) return;
    if(st == dr)
    {
        d[nod] -= val;
        return;
    }
    achita(nod * 2, st, (st + dr) / 2, i, val);
    achita(nod * 2 + 1, (st + dr) / 2 + 1, dr, i, val);
    d[nod] = d[nod * 2] + d[nod * 2 + 1];
}

int suma(int nod, int st, int dr, int i, int j)
{
    if(dr < i || st > j) return 0;
    if(i <= st && dr <= j) return d[nod];
    return suma(nod * 2, st, (st + dr) / 2, i, j) + suma (nod * 2 + 1, (st + dr) / 2 + 1, dr, i, j);
}

int main()
{int n, m, i, x, y, o;
ifstream f("datorii.in");
ofstream g("datorii.out");
f >> n >> m;
for(i = 1; i <= n; i++)
    f >> a[i];
creare(1, 1, n);
for(i = 0; i < m; i++)
{
    f >> o >> x >> y;
    if(o) g << suma(1, 1, n, x, y) << '\n';
    else achita(1, 1, n, x, y);
}
f.close();
g.close();
return 0;
}