Cod sursa(job #3194631)

Utilizator AlexandraRusRus Alexandra Maria AlexandraRus Data 18 ianuarie 2024 19:31:14
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.28 kb
#include <bits/stdc++.h>

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

int v[100001], arbint[400001], n, m;
int c, a, b;

int query(int nod, int st, int dr, int intrst, int intrdr)
{
    if(st==intrst && dr==intrdr) return arbint[nod];
    int mij=(intrst+intrdr)/2;
    if(st<=mij && mij<dr)
        return query(nod*2, st, mij, intrst, mij)+query(nod*2+1, mij+1, dr, mij+1, intrdr);
    if(mij<st)
        return query(2*nod+1, st, dr, mij+1, intrdr);
    if(mij>=dr)
        return query(2*nod, st, dr, intrst, mij);
}

void update(int nod, int st, int dr, int p)
{
    if(st==dr)
    {
        arbint[nod]=v[p];
        return;
    }
    int mij=(st+dr)/2;
    if(mij>=p)
        update(2*nod, st, mij, p);
    else
        update(2*nod+1, mij+1, dr, p);
    arbint[nod]=arbint[2*nod]+arbint[2*nod+1];
}

void solve()
{
    for(int i=1;i<=n;i++)
    {
        update(1, 1, n, i);
    }
    for(int i=1;i<=m;i++)
    {
        fin>>c>>a>>b;
        if(c==1)
            fout<<query(1, a, b, 1, n)<<"\n";
        else
        {
            v[a]=v[a]-b;
            update(1, 1, n, a);
        }
    }
}

int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
        fin>>v[i];
    solve();
    return 0;
}