Cod sursa(job #2775540)

Utilizator TraianQTraianQ TraianQ Data 16 septembrie 2021 10:25:04
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.34 kb
#include <fstream>
using namespace std;
int t[45001],v[15001],pos,val,Start,End;
void build(int node,int st,int dr)
{
    if(st==dr)
        t[node]=v[st];
    else
    {
        int mij=(st+dr)/2;
        build(2*node,st,mij);
        build(2*node+1,mij+1,dr);
        t[node]=t[2*node]+t[2*node+1];
    }
}
void update(int node,int st,int dr)
{
    if(st==dr)
        t[node]-=val;
    else
    {
        int mij=(st+dr)/2;
        if(pos<=mij)
            update(2*node,st,mij);
        else
            update(2*node+1,mij+1,dr);
        t[node]=t[2*node]+t[2*node+1];
    }
}
void query(int node,int st,int dr)
{
    if(Start <= st && dr <= End)
        val+=t[node];
    else
    {
        int mij=(st+dr)/2;
        if(Start<=mij)
            query(2*node,st,mij);
        if(End > mij)
            query(2*node+1,mij+1,dr);
    }
}
int main()
{
    ifstream cin("datorii.in");
    ofstream cout("datorii.out");
    int n,m,tip,a,b;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>v[i];
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        cin>>tip>>a>>b;
        if(tip==0)
        {
            pos=a,val=b;
            update(1,1,n);
        }
        else
        {
            val=0,Start=a, End=b;
            query(1,1,n);
            cout<<val<<'\n';
        }
    }
    return 0;
}