Cod sursa(job #951516)

Utilizator Andrei1998Andrei Constantinescu Andrei1998 Data 20 mai 2013 19:59:28
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.96 kb
//Arbori de intervale
// -> update simplu (pe element) 
// -> query de suma
// Complexitate totala O(m*logn)
#include <fstream>

using namespace std;

//Un nod al arborelui
struct nod
{
   int st,dr,sum;
   nod()
   {
      st=0;
      dr=0;
      sum=0;     
   }   
}arb[32768];
int v[15005],aux;

//Construieste arborele de intervale
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;
}

//Update pe arbore
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;
}

//Query pe interval
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(dr<=mijl)
     query(st,dr,poz<<1);
   else if(st>mijl)
     query(st,dr,(poz<<1)+1);
   else
     query(st,mijl,poz<<1),query(mijl+1,dr,(poz<<1)+1);
}

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