Cod sursa(job #1050105)

Utilizator apopeid15Apopei Daniel apopeid15 Data 8 decembrie 2013 10:41:42
Problema Factorial Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.72 kb
#include <fstream>
 
using namespace std;
 
struct nod
{
   int st,dr,sum;
   nod()
   {
      st=0;
      dr=0;
      sum=0;    
   }  
}arb[32768];
int v[15005],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(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);
}
 
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;
}