Cod sursa(job #2592328)

Utilizator maramihaliMara Mihali maramihali Data 1 aprilie 2020 16:18:56
Problema Datorii Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.33 kb
#include <bits/stdc++.h>

using namespace std;

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

const int MAX = 1 << 15;

int t[MAX], val, poz, a, b;

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

int interogare(int p, int st, int dr)
{
    if(a <= st && dr <= b)
    {
        return t[p];
    }
    int m = (st + dr) / 2;
    int maxst = 0;
    int maxdr = 0;
    if(a <= m)
    {
        maxst = interogare(2 * p, st, m);
    }
    if(b > m)
    {
        maxdr = interogare(2 * p + 1, m + 1, dr);
    }
    return maxst + maxdr;
}

int main()
{
    int n, m, op, x, y;
    in >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        in >> val;
        poz = i;
        actualizare(1, 1, n);
    }
    for(int i = 1; i <= m; i++)
    {
        in >> op;
        if(op == 0)
        {
            in >> poz >> val;
            val = -val;
            actualizare(1, 1, n);
        }
        else
        {
            in >> a >> b;
            out << interogare(1, 1, n) << "\n";
        }
    }
    return 0;
}