Cod sursa(job #2917808)

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

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



struct segtree
{
    vector <int> tree;

    int tsize, intersize;
    void init(int n)
    {
        tsize = 1;
        while(tsize < n)
            tsize <<= 1;
        intersize = tsize;
        tsize <<= 1;
        tree.assign(tsize, 0);
    }
    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)
    {
        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;

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

};



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

    fin >> n >> q;

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

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


    return 0;
}