Cod sursa(job #522227)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 14 ianuarie 2011 16:25:26
Problema Arbori indexati binar Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.83 kb
#include<stdio.h>
#include<math.h>
#include<iostream.h>
#include<fstream.h>
#define N 100001
long n,m,y,z,i,j,step,a[N];
int x,poz;
unsigned long long c[N],b[N],p;

unsigned long long suma(unsigned long long c[N],long y,long z,long n)
{long j=y-1,i=z;
int poz=0;
unsigned long long s1=0,s2=0;
while(i>0)
       {s1+=c[i];
       while((i&(long)pow(2,poz))==0)
             poz++;
       i-=(long)pow(2,poz);
       poz++;}
poz=0;
while(j>0)
       {s2+=c[j];
       while((j&(long)pow(2,poz))==0)
             poz++;
       j-=(long)pow(2,poz);
       poz++;}
return s1-s2;}
       
long pozitie(unsigned long long c[N],unsigned long long p,long n)
{long step,i;
unsigned long k=p;
for(step=1;step<n;step<<=1);
step>>=1;
if(suma(c,1,n,n)==k)
       return n;
for(i=0;step!=0;step>>=1)
if(i+step<n&&c[i+step]<=k)
       {i+=step;
       k-=c[i];}
if(k==0)
       return i;
else
       return -1;}

int main()
{fstream f1("aib.in",ios::in);
fstream f2("aib.out",ios::out);
f1>>n>>m;
b[0]=0;
for(i=1;i<=n;i++)
       {f1>>a[i];
       b[i]=b[i-1]+a[i];
       j=i;
       poz=0;
       while((j&(long)pow(2,poz))==0)
              poz++;
       j=j-(long)pow(2,poz);
       c[i]=b[i]-b[j];}
for(i=1;i<=m;i++)
       {f1>>x;
       if(x==2)
              {f1>>p;
              f2<<pozitie(c,p,n)<<endl;}
       else
              if(x==0)
                      {f1>>y>>z;
                      poz=0;
                      while(y<=n)
                             {c[y]+=z;
                             while((y&(long)pow(2,poz))==0)
                                       poz++;
                             y+=(long)pow(2,poz);
                             poz++;}}
              else
                      {f1>>y>>z;
                      f2<<suma(c,y,z,n)<<endl;}}
f1.close();
f2.close();
return 0;}