Cod sursa(job #3259479)

Utilizator tonealexandruTone Alexandru tonealexandru Data 26 noiembrie 2024 15:15:09
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.61 kb
#include <bits/stdc++.h>

using namespace std;
int v[100005];
int aib[400005];

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

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

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

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

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

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

int query(int node, int st, int dr, int qst, int qdr)
{
    if(qst <= st && dr <= qdr)
        return aib[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 i=1; i<=m; i++)
    {
        cin>>cer>>a>>b;
        if(cer == 0)
        {
            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;
}