Cod sursa(job #2902113)

Utilizator kanyjmkSabau Eduard kanyjmk Data 15 mai 2022 17:07:58
Problema Datorii Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int arbint[60006], n, m;
void add_value(int node_poz, int left, int right, int poz, int x)
{
    if(left == right)
        arbint[node_poz] = x;
    else
    {
        int m = (left+right) / 2;
        if(poz <= m)
            add_value(node_poz*2, left, m, poz, x);
        else add_value(node_poz*2 + 1, m+1, right, poz, x);
        arbint[node_poz] = arbint[node_poz*2] + arbint[node_poz*2 + 1];
    }
}
void change_value(int node_poz, int left, int right, int poz, int x)
{
    if(left == right)
        arbint[node_poz] -= x;
    else
    {
        int m = (left+right) / 2;
        if(poz <= m)
            change_value(node_poz*2, left, m, poz, x);
        else change_value(node_poz*2 + 1, m+1, right, poz, x);
        arbint[node_poz] = arbint[node_poz*2] + arbint[node_poz*2 + 1];
    }
}
int find_sum(int node_poz, int left, int right, int a, int b)
{
    if(a > right || b < left)
        return 0;

    if(a <= left && right <= b)
        return arbint[node_poz];

    int m = (left+right) / 2;
    return find_sum(node_poz*2, left, m, a, b) + find_sum(node_poz*2+1, m+1, right, a, b);
}
int main()
{
    int a, b, x, op;
    fin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        fin >> x;
        add_value(1, 1, n, i, x);
    }
    for(int i = 1; i <= m; i++)
    {
        fin >> op >> a >> b;
        if(op == 1)
            fout << find_sum(1, 1, n, a, b) << '\n';
        else change_value(1, 1, n, a, b);
    }

}