Cod sursa(job #2874453)

Utilizator NeganAlex Mihalcea Negan Data 19 martie 2022 14:37:26
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.12 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin ("datorii.in");
ofstream fout ("datorii.out");

int n, m, a[70006];


void Update(int p, int val)
{
    int t;
    a[p] -= val;
    while (p >= 1)
    {
        t = p / 2;
        a[t] -= val;
        p = t;
    }
}

int Query(int root, int x, int y, int st, int dr)
{
    if (x == st && y == dr)
        return a[root];
    int m = (st + dr) / 2;
    if (y <= m)
        return Query(2 * root, x, y, st, m);
    if (x >= m + 1)
        return Query(2 * root + 1, x , y, m + 1, dr);
    return Query(2 * root, x, m, st, m) + Query(2 * root + 1, m + 1, y, m + 1, dr);
}

int main()
{
    fin >> n >> m;
    int k;
    for (k = 1; k < n; k *= 2)
        ;
    for (int i = k; i < k + n; i++)
        fin >> a[i];
    n= k;
    for (int i = n - 1; i > 0; i--)
        a[i] = a[2 * i] + a[2 * i + 1];
    while (m--)
    {
        int task, x , y;
        fin >> task >> x >> y;
        if (task == 0)
        {
            Update(n - 1 + x, y);
        }
        else fout << Query (1, x, y, 1, n) << "\n";
    }
    return 0;
}