Cod sursa(job #2900735)

Utilizator simaderalSimader Albert simaderal Data 12 mai 2022 04:06:44
Problema Datorii Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.5 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <string>
using namespace std;
#define INT_MAX 2147483647;
int v[400040];
int res = 0;
void update(int nod, int st, int dr, int pos, int val, bool modify = false)
{
    if (st == dr)
    {
        if (modify)
            v[nod] -= val; // s-a mai platit din rata
        else
            v[nod] = val; 
    }
    else
    {
        int mij = (st + dr) / 2;
        if (mij >= pos)
            update(nod * 2, st, mij, pos, val, modify);
        else
            update(nod * 2 + 1, mij + 1, dr, pos, val, modify);
        v[nod] = v[nod * 2] + v[nod * 2 + 1];
    }
}
void call(int nod, int st, int dr, int a, int b) {
    if (a <= st && dr <= b) {
        res += v[nod];
        return;
    }
    else {
        int mij = (st + dr) / 2;
        if (mij >= a)
            call(nod * 2, st, mij, a, b);
        if (mij + 1 <= b)
            call(nod * 2 + 1, mij + 1, dr, a, b);
    }
}
int main(void)
{
    int n, m, x, c, a, b;
    ifstream f("datorii.in");
    ofstream g("datorii.out");

    f >> n >> m;
    for (int i = 1; i <= n; ++i) {
        f >> x;
        //int nod, int st, int dr, int pos, int val
        update(1, 1, n, i, x);
    }
    for (int i = 1; i <= m; ++i)
    {
        f >> c >> a >> b;
        if (c == 1) {
            res = 0;
            call(1, 1, n, a, b);
            g << res << '\n';
        }
        else
        {
            update(1, 1, n, a, b, 1);
        }
    }
}