Pagini recente » Cod sursa (job #1693409) | Cod sursa (job #2317616) | Cod sursa (job #1915629) | Cod sursa (job #1376097) | Cod sursa (job #2485955)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n , aib[100001] , m;
void Update(int p , int x)
{
while(p <= n)
{
aib[p] += x;
/// aflu cati biti de 0 sunt la finalul lui p (2 ^ nrbiti)
int i = 1; /// 2^0
while(p % (i * 2) == 0)
{
i = i * 2;
}
p = p + i;
}
}
/// suma de la a[1] la a[p]
int Suma(int p)
{
int S = 0;
while(p > 0)
{
S += aib[p];
p -= (p & (-p));
/// sau acelasi lucru facut la Update
}
return S;
}
/// cauta cea mai din stanga pozitie poz
/// cu suma a[1] + .... a[poz] = s;
int CautBin(int p , int s)
{
int st , dr , poz , mij;
int suma;
st = 1;
dr = p;
poz = -1;
while(st <= dr)
{
mij = (st + dr) / 2;
suma = Suma(mij);
if(suma == s)
{
poz = mij;
dr = mij - 1;
}
else if(suma > s)
{
dr = mij - 1;
}
else st = mij + 1;
}
return poz;
}
int main()
{
int i , op , a , b , x;
fin >> n >> m;
for(i = 1 ; i <= n ; ++i)
{
fin >> x;
Update(i , x);
}
while(m--)
{
fin >> op;
if(op == 0)
{
fin >> a >> b;
Update(a , b);
}
else if(op == 1)
{
fin >> a >> b;
fout << (Suma(b) - Suma(a - 1)) << "\n";
}
else
{
fin >> a;
fout << CautBin(n , a) << "\n";
}
}
return 0;
}