Cod sursa(job #2579854)

Utilizator Alex100Alexandru Mihai Alex100 Data 12 martie 2020 21:50:43
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.16 kb
#include <bits/stdc++.h>
#define MAX 61000
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int arb[MAX];
void update(int node,int st,int dr,int pos,int val);
int query(int node,int st,int dr,int x,int y);
int main()
{
    int n,m,x,y,i;
    bool ce;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>x;
        update(1,1,n,i,x);
    }
    for(i=1;i<=m;i++)
    {
        fin>>ce>>x>>y;
        if(!ce)
            update(1,1,n,x,-y);
        else
            fout<<query(1,1,n,x,y)<<'\n';
    }
    return 0;
}
void update(int node,int st,int dr,int pos,int val)
{
    int mij;
    if(st==dr)
    {
        arb[node]+=val;
        return;
    }
    mij=(st+dr)/2;
    if(pos<=mij)
       update(2*node,st,mij,pos,val);
    else
       update(2*node+1,mij+1,dr,pos,val);
    arb[node]=arb[2*node]+arb[2*node+1];

}
int query(int node,int st,int dr,int x,int y)
{
    int mij,ans_l=0,ans_r=0;
    if(x<=st&&dr<=y)
        return arb[node];
    mij=(st+dr)/2;
    if(x<=mij)
        ans_l=query(2*node,st,mij,x,y);
    if(y>mij)
        ans_r=query(2*node+1,mij+1,dr,x,y);
    return ans_l+ans_r;
}