Cod sursa(job #2917810)

Utilizator Luca_CristianZamfir Luca-Cristian Luca_Cristian Data 7 august 2022 19:51:00
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <fstream>
#include <vector>

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


vector <int> tree(1000001);
struct segtree
{
    void setare(int i, int v, int node, int lx, int rx)
    {
        if(rx - lx == 1)
        {
            tree[node] = v;
            return ;
        }
        int mij = (lx + rx) / 2;
        if(mij > i)
            setare(i, v, 2 * node + 1, lx, mij);
        else
            setare(i, v, 2 * node + 2, mij, rx);
        tree[node] = tree[2 * node + 1] + tree[2 * node + 2];
    }
    void setare(int i, int v, int intersize)
    {
        setare(i, v, 0, 0, intersize);
    }
    long long query(int l, int r, int node, int lx, int rx)
    {
        if(rx <= l || lx >= r)
            return 0;
        if(l <= lx && rx <= r)
            return tree[node];

        int mij= (lx + rx) / 2;
        return query(l, r, 2 * node + 1, lx, mij) + query(l, r, 2 * node + 2, mij, rx);
    }
    long long query(int l, int r,  int intersize)
    {
        return query(l, r, 0, 0, intersize);
    }

};



int main()
{
    int n, i, q;

    fin >> n >> q;

    segtree sg;
    vector <int> v(n);
    for(i = 0; i < n; i++)
        fin >> v[i], sg.setare(i, v[i], n);

    for(i = 0; i < q; i++)
    {
        int tip, a, b;
        fin >> tip >> a >> b;
        if(tip == 0)
        {
            a--;
            sg.setare(a, v[a] - b, n);
        }
        else
        {
            a--;
            fout << sg.query(a, b, n) << '\n';
        }
    }


    return 0;
}