Cod sursa(job #3261228)

Utilizator tonealexandruTone Alexandru tonealexandru Data 4 decembrie 2024 20:15:56
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.67 kb
#include <bits/stdc++.h>

using namespace std;
int v[100005], aint[400005];

void build(int node, int st, int dr)
{
    if(st == dr)
        aint[node] = v[st];
    else
    {
        int mij = (st + dr) / 2;

        build(node*2, st, mij);
        build(node*2+1, mij+1, dr);

        aint[node] = aint[node*2] + aint[node*2+1];
    }
}

void update(int node, int st, int dr, int poz, int val)
{
    if(st == dr)
        aint[node] = val;
    else
    {
        int mij = (st + dr) / 2;

        if(mij >= poz)
            update(node*2, st, mij, poz, val);
        else
            update(node*2+1, mij+1, dr, poz, val);

        aint[node] = aint[node*2] + aint[node*2+1];
    }
}

int query(int node, int st, int dr, int qst, int qdr)
{
    if(qst <= st && dr <= qdr)
        return aint[node];
    else
    {
        int mij = (st + dr) / 2;

        if(qdr <= mij)
            return query(node*2, st, mij, qst, qdr);
        if(mij+1 <= qst)
            return query(node*2+1, mij+1, dr, qst, qdr);

        return (query(node*2, st, mij, qst, qdr) + query(node*2+1, mij+1, dr, qst, qdr));
    }
}

int main()
{
    ifstream cin("datorii.in");
    ofstream cout("datorii.out");
    int n, m;
    cin>>n>>m;

    for(int i=1; i<=n; i++)
        cin>>v[i];

    build(1, 1, n);

    int cer, a, b;
    for(int j=1; j<=m; j++)
    {
        cin>>cer>>a>>b;
        //cout<<"KOK";

        if(cer == 0)
        {
            //cout<<"KOK2";
            int val = query(1, 1, n, a, a);
            update(1, 1, n, a, val - b);
        }
        else if(cer == 1) cout<<query(1, 1, n, a, b)<<'\n';
    }


    return 0;
}