Cod sursa(job #2421790)

Utilizator baranceanuBaranceanu Vlad baranceanu Data 16 mai 2019 09:58:56
Problema Datorii Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <bits/stdc++.h>
#define NMAX 15005
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");

int aint[4*NMAX];
int V[NMAX];
int n,m;

void build(int nod,int st,int dr);
int rezultat(int nod,int st,int dr,int a,int b);
void update(int nod,int st,int dr,int val,int poz);
int main()
{int i,cer,a,b;
 fin>>n>>m;
 for(i=1;i<=n;i++)
    fin>>V[i];
 build(1,1,n);
 for(i=1;i<=m;i++)
    {fin>>cer>>a>>b;
     if(cer==1)
       fout<<rezultat(1,1,n,a,b)<<'\n';
     if(cer==0)
        update(1,1,n,b,a);
    }
}
void build(int nod,int st,int dr)
{int i;
 if(st==dr)
    {aint[nod]=V[st];
     return;
    }
 int mij=(st+dr)/2;
 build(2*nod,st,mij);
 build(2*nod+1,mij+1,dr);
 aint[nod]=aint[2*nod]+aint[2*nod+1];
}
int rezultat(int nod,int st,int dr,int a,int b)
{if(st>b || dr<a)
   return 0;
 if(st>=a && dr<=b)
   return aint[nod];
 int mij=(st+dr)/2;
 return rezultat(2*nod,st,mij,a,b)+rezultat(2*nod+1,mij+1,dr,a,b);
}
void update(int nod,int st,int dr,int val,int poz)
{if(st==dr && st==poz)
   {aint[nod]-=val;
    return;
   }
 int mij=(st+dr)/2;
 if(mij<poz)
    update(2*nod+1,mij+1,dr,val,poz);
    else
    update(2*nod,st,mij,val,poz);
 aint[nod]=aint[2*nod]+aint[2*nod+1];
}