Cod sursa(job #2426883)

Utilizator AlexandruBrezuleanuAlexandruBrezuleanu AlexandruBrezuleanu Data 29 mai 2019 20:28:46
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <bits/stdc++.h>
#define MAX 60100
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int v[MAX];
int n,m;
void read();
int query(int node,int x,int y,int st,int dr);
void update(int node,int val,int pos,int st,int dr);
void scad(int node,int val,int pos,int st,int dr);
int main()
{
    read();
    return 0;
}
void read()
{
    int n,m,i,val,x,y;
    bool ce;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>val;
        update(1,val,i,1,n);
    }
    for(i=1;i<=m;i++)
    {
        fin>>ce>>x>>y;
        if(!ce)
            scad(1,y,x,1,n);
        else
            fout<<query(1,x,y,1,n)<<'\n';
    }
}
int query(int node,int x,int y,int st,int dr)
{
    int mij,ans_l=0,ans_r=0;
    if(x<=st&&dr<=y)
        return v[node];
    mij=(st+dr)/2;
    if(x<=mij)
        ans_l=query(2*node,x,y,st,mij);
    if(y>mij)
        ans_r=query(2*node+1,x,y,mij+1,dr);
    return ans_l+ans_r;
}
void update(int node,int val,int pos,int st,int dr)
{
    int mij;
    if(st==dr)
        {v[node]=val;
        return;}
    mij=(st+dr)/2;
    if(pos<=mij)
        update(2*node,val,pos,st,mij);
    else
        update(2*node+1,val,pos,mij+1,dr);
    v[node]=v[2*node]+v[2*node+1];
}
void scad(int node,int val,int pos,int st,int dr)
{
    int mij;
    if(st==dr)
        {v[node]-=val;
        return;}
    mij=(st+dr)/2;
    if(pos<=mij)
        update(2*node,val,pos,st,mij);
    else
        update(2*node+1,val,pos,mij+1,dr);
    v[node]-=val;
}