Cod sursa(job #951495)

Utilizator Andrei1998Andrei Constantinescu Andrei1998 Data 20 mai 2013 19:12:35
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.75 kb
#include <fstream>

using namespace std;

struct nod
{
   int st,dr,sum;
   nod()
   {
      st=0;
      dr=0;
      sum=0;     
   }   
}arb[382768];
int v[159005],aux;

void build(int st,int dr,int poz)
{
   arb[poz].st=st;
   arb[poz].dr=dr;
   
   if(st==dr)
   {
      arb[poz].sum=v[st];
      return;
   }
   int mijl=(st+dr)>>1;
   build(st,mijl,poz<<1);
   build(mijl+1,dr,(poz<<1)+1);
   arb[poz].sum=arb[poz<<1].sum+arb[(poz<<1)+1].sum;
}

void update(int unde,int val,int poz)
{
  if(arb[poz].st==arb[poz].dr && arb[poz].st==unde)
  {
     arb[poz].sum-=val;
     v[unde]-=val;
     return;
  }
  int mijl=(arb[poz].st+arb[poz].dr)>>1;
  if(unde<=mijl)
    update(unde,val,poz<<1);
  else if(unde>mijl)
    update(unde,val,(poz<<1)+1);
  arb[poz].sum=arb[poz<<1].sum+arb[(poz<<1)+1].sum;
}

void query(int st,int dr,int poz)
{
   if((arb[poz].dr<st) || (arb[poz].st>dr))
     return;
   if((arb[poz].st>=st) && (arb[poz].dr<=dr))
   {
      aux+=arb[poz].sum;
      return;
   }
   int mijl=(arb[poz].st+arb[poz].dr)>>1;
   if(arb[poz].dr<=mijl)
     query(st,dr,poz<<1);
   else if(arb[poz].st>mijl)
     query(st,dr,(poz<<1)+1);
   else
     query(st,mijl,poz<<1),query(mijl+1,dr,(poz<<1)+1);
}

int main()
{
    ifstream cin("datorii.in");
    ofstream cout("datorii.out");
    
    int n,m,i,cod,a,b;
    cin>>n>>m;
    for(i=1;i<=n;i++)
      cin>>v[i];
    build(1,n,1);
      
    for(i=0;i<m;i++)   
    {
       cin>>cod>>a>>b;
       if(cod)
       {
          aux=0;
          query(a,b,1);
          cout<<aux<<'\n';
       }                  
       else    
          update(a,b,1);
    }   
    cin.close();
    cout.close();
    
    //system("PAUSE");
    return 0;
}