Cod sursa(job #2401385)

Utilizator Mihnea_BranzeuMihnea Branzeu Mihnea_Branzeu Data 9 aprilie 2019 17:43:59
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.98 kb
#include <fstream>

using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int arb[1<<18],a,b,poz,val;

int query(int p,int st,int dr)
{
    if(a<=st && dr<=b)
      return arb[p];
    int m=(st+dr)/2,m1=0,m2=0;
    if(a<=m)
       m1=query(2*p,st,m);
    if(b>m)
       m2=query(2*p+1,m+1,dr);
    return m1+m2;
}

void update(int p,int st,int dr)
{
    if(st==dr)
    {
        arb[p]-=val;
        return;
    }
    int m=(st+dr)/2;
    if(poz<=m)
        update(2*p,st,m);
    else
            update(2*p+1,m+1,dr);
    arb[p]=arb[2*p]+arb[2*p+1];
}
int main()
{
    int n,nrop,boi;
    fin>>n>>nrop;
    for(int i=1;i<=n;i++)
      {
         fin>>val;
         val=(-1)*val;
         poz=i;
         update(1,1,n);
      }
    for(int i=1;i<=nrop;i++)
    {
       fin>>boi;
       if(boi==1)
       {
          fin>>a>>b;
          fout<<query(1,1,n)<<"\n";
       }
       else
       {
          fin>>poz>>val;
          update(1,1,n);
       }
    }
    return 0;
}