Cod sursa(job #2951498)

Utilizator MacraAlexandruMacra Alexandru MacraAlexandru Data 6 decembrie 2022 17:24:54
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.46 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int a[100005], n, m, sg[400005];
void build(int nod, int st, int dr)
{
    if(st==dr)sg[nod]=a[st];
    else
    {
        int mij=(st+dr)/2;
        build(2*nod, st, mij);
        build(2*nod+1, mij+1, dr);
        sg[nod]=sg[nod*2]+ sg[nod*2+1];
    }
}
void update(int nod, int st, int dr, int pos, int nv)
{
    if(st==dr)sg[nod]=a[st]-nv;
    else
    {
        int mij=(st+dr)/2;
        if(pos<mij)update(nod*2, st, mij, pos, nv);
        else update(2*nod+1, mij+1, dr, pos, nv);

        sg[nod]=sg[nod*2]+ sg[2*nod+1];
    }
}
int query(int nod, int st, int dr, int qst, int qdr)
{
    if(qst<=st && dr<=qdr)return sg[nod];
    else
    {
        int mij=(st+dr)/2;
        if(qdr<=mij)
            return query(nod*2, st, mij, qst, qdr);
        if(mij+1<=qst)
            return query(nod*2+1, mij+1, dr, qst, qdr);

        return query(2*nod, st, mij, qst, qdr)+ query(nod*2+1, mij+1, dr, qst, qdr);
    }
}
int main()
{
    f>>n>>m;
    for(int i=1; i<=n; i++)f>>a[i];
    build(1, 1, n);
    for(int i=1; i<=m; i++)
    {
        int c;
        f>>c;
        if(c==0)
        {
            int t, v;
            f>>t>>v;
            update(1, 1, n, t, v);
        }
        else
        {
            int p, q;
            f>>p>>q;
            g<<query(1, 1, n, p, q)<<endl;
        }
    }
    return 0;
}