Cod sursa(job #3215551)

Utilizator Kawaiimeatball13Russu Mihaela Kawaiimeatball13 Data 15 martie 2024 09:50:23
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <iostream>
#include <fstream>
using namespace std;

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

int n, m;
int a[15001];
int arb[60001];

void update(int st, int dr, int tata, int poz, int val)
{
    if(st == dr)
    {
        arb[tata] = val;
        return;
    }

    int mij = (st + dr) / 2;
    if(poz <= mij)
        update(st, mij, tata * 2, poz, val);
    else
        update(mij + 1, dr, tata * 2 + 1, poz, val);
    arb[tata] = arb[tata * 2] + arb[tata * 2 + 1];
}

int query(int a, int b, int st, int dr, int tata)
{
    if(st >= a && dr <= b)
        return arb[tata];
    int s = 0;
    int mij = (st + dr) / 2;
    if(a <= mij)
        s += query(a, b, st, mij, tata * 2);
    if(b > mij)
        s += query(a, b, mij + 1, dr, tata * 2 + 1);
    return s;
}

void citire()
{
    fin >> n >> m;
    for(int i = 1; i <= n; ++i)
    {
        fin >> a[i];
        update(1, n, 1, i, a[i]);
    }
}

int main()
{
    citire();

    int cerinta, x, y;
    for(int i = 0 ; i < m; ++i)
    {
        fin >> cerinta >> x >> y;
        if(cerinta == 0)
            update(1, n, 1, x, a[x] - y);
        else
            fout << query(x, y, 1, n, 1) << '\n';
    }
    return 0;
}