Pagini recente » Cod sursa (job #2426134) | Cod sursa (job #3129696) | Cod sursa (job #3191793) | Cod sursa (job #795654) | Cod sursa (job #1872450)
#include<fstream>
#include<cstring>
using namespace std;
ifstream f("aib.in");
ofstream g("aib.out");
int n,op,cer,poz,val,nr,poz1,poz2,i;
int tree[100001];
int suma,put;
void update(int poz,int val)//incrementez cu val pozitia poz
{
put=0;
while(poz<=n)
{
tree[poz]+=val;
while(!(poz&(1<<put)))
put++;
poz+=(1<<put);
put++;
}
}
int query(int poz)//calculez suma primelor poz elemente
{
put=0,suma=0;
while(poz>0)
{
suma+=tree[poz];
while(!(poz&(1<<put)))
put++;
poz-=(1<<put);
put++;
}
return suma;
}
int caut(int val)//caut pozitia i pentru care suma primelor i elemente ale sirului au suma egala cu val
{
int i,step=1;
while(step<n)
step<<=1;
for(i=0;step;step>>=1)
if(i+step<=n&&val>=tree[i+step])
{
i+=step;
val-=tree[i];
if(!val)
return i;
}
return -1;
}
int main()
{
f>>n>>cer;
for(i=1;i<=n;i++)
{
f>>nr;
update(i,nr);//adaug in arbore elemente
}
while(cer--)
{
f>>op;
if(op==2)//trebuie sa caut pozitia minima k pentru care suma primelor k valori este val
{
f>>val;
g<<caut(val)<<"\n";
}
else if(op==1)//suma elementelor de pe intervalul [poz1,poz2]
{
f>>poz1>>poz2;
g<<query(poz2)-query(poz1-1)<<"\n";
}
else if(op==0)//trebuie sa adaug elementul de pe poz valoare val
{
f>>poz>>val;
update(poz,val);
}
}
return 0;
}