Cod sursa(job #2774346)

Utilizator CelestinNegraru Celestin Celestin Data 11 septembrie 2021 12:04:10
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.66 kb
#include <iostream>
#include <fstream>
#define maxi 15000
#define LSON(x) 2*x
#define RSON(x) 2*x+1
#define MID(x,y) (x+y)/2
using namespace std;
ifstream f;
ofstream g;
int v[14*maxi],a[maxi+5],n,m,x,op,c,b;
void BUILD(int nod,int st,int dr)
{
    if(st>dr)
        return;
    if(st==dr)
    {
        v[nod]=a[st];
        return;
    }
    BUILD(LSON(nod),st,MID(st,dr));
    BUILD(RSON(nod),MID(st,dr)+1,dr);
    v[nod]=v[RSON(nod)]+v[LSON(nod)];
}
void UPDATE(int nod,int st,int dr,int poz,int val)
{
    if(st>dr)
        return;
    if(st==dr)
    {
        v[nod]-=val;
        return;
    }
    if(poz<=MID(st,dr))
        UPDATE(LSON(nod),st,MID(st,dr),poz,val);
    else UPDATE(RSON(nod),MID(st,dr)+1,dr,poz,val);
    v[nod]=v[LSON(nod)]+v[RSON(nod)];
}
void QUERY(int nod,int st,int dr,int a,int b,int &suma)
{
    if(st>dr)
        return;
    if(st>=a&&dr<=b)
    {
        suma+=v[nod];
        return;
    }
    if(st>b||dr<a)
        return;
    if(a<=MID(st,dr))
        QUERY(LSON(nod),st,MID(st,dr),a,b,suma);
    if(b>MID(st,dr))
        QUERY(RSON(nod),MID(st,dr)+1,dr,a,b,suma);
}
void SOLVE()
{
    f.open("datorii.in",ios::in);
    g.open("datorii.out",ios::out);
    f>>n>>m;
    for(int i=1;i<=n;i++)
        {
            f>>a[i];
        }
    BUILD(1,1,n);
    for(int i=1;i<=m;i++)
    {
        f>>op>>b>>c;
        if(op==0)
        {
            UPDATE(1,1,n,b,c);
        }
        else{
            int summy=0;
            QUERY(1,1,n,b,c,summy);
            g<<summy<<'\n';
        }
    }
    f.close();
    g.close();
    return;
}
int main()
{
    SOLVE();
    return 0;
}