Cod sursa(job #3288767)

Utilizator raresmihai1234Rares Mihai raresmihai1234 Data 24 martie 2025 10:47:57
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.49 kb
#include <fstream>
#define N 15005
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int n,k,v[N],a[N*5],aa,bb,xx,xy,mij,ans,a2,vt;
void build(int nod,int st,int dr)
{
    if (st==dr)
    {
        a[nod]=v[st];
    }
    else
    {
        int mij=(st+dr)>>1;
        build(2*nod,st,mij);
        build(2*nod+1,mij+1,dr);
        a[nod]=a[nod*2]+a[nod*2+1];
    }
}
void update(int nod,int st,int dr,int p,int sum)
{
    if (st==dr)
    {
        v[st]-=sum;
        a[nod]=v[st];
    }
    else
    {
        int mij=(st+dr)>>1;
        if (p<=mij)
        {
            update(2*nod,st,mij,p,sum);
        }
        else if (p>=mij+1)
        {
            update(2*nod+1,mij+1,dr,p,sum);
        }
        a[nod]=a[nod*2]+a[nod*2+1];
    }
}
void query(int nod,int st,int dr,int a2,int b)
{
    if (st>=a2&&dr<=b)
    {
        ans+=a[nod];
    }
    else
    {
        int mij=(st+dr)>>1;
        if (a2<=mij) query(2*nod,st,mij,a2,b);
        if (b>=mij+1) query(2*nod+1,mij+1,dr,a2,b);
    }
}
int main()
{
    fin>>n>>k;
    for (int i=1;i<=n;i++)
    {
        fin>>v[i];
    }
    build(1,1,n);
    for (int i=1;i<=k;i++)
    {
        fin>>vt;
        if (vt==0)
        {
            fin>>xy>>xx;
            update(1,1,n,xy,xx);
        }
        else
        {
            fin>>aa>>bb;
            ans=0;
            query(1,1,n,aa,bb);
            fout<<ans<<'\n';
        }
    }
    return 0;
}