Pagini recente » Cod sursa (job #2045609) | Cod sursa (job #1621589) | Cod sursa (job #420192) | Cod sursa (job #2332670) | Cod sursa (job #2036544)
#include <fstream>
#include <vector>
using namespace std;
ifstream in("aib.in");
ofstream out("aib.out");
#define MAXN 100000
int v[MAXN], AIB[MAXN + 1];
int n;
void Update(int x, int val)
{
for (; x <= n; x += x & (-x))
{
AIB[x] += val;
}
}
int Query(int x)
{
int sum = 0;
for (; x; x -= x & (-x))
{
sum += AIB[x];
}
return sum;
}
int SearchSum(int sum)
{
int mask = 1;
while ((mask << 1) <= n) mask <<= 1;
int poz = 0;
while (mask && poz <= n)
{
int tpoz = poz | mask;
if (tpoz <= n)
{
if (AIB[tpoz] == sum)
{
return tpoz;
}
else if (AIB[tpoz] < sum)
{
sum -= AIB[tpoz];
poz = tpoz;
}
}
mask >>= 1;
}
if (sum == 0)
return poz;
return -1;
}
int main()
{
int N, M;
in >> N >> M;
for (int i = 0; i < N; i++)
{
in >> v[i];
Update(i + 1, v[i]);
}
while (M--)
{
int opt, a, b;
in >> opt;
switch (opt)
{
case 0:
in >> a >> b;
Update(a, b);
break;
case 1:
in >> a >> b;
out << Query(b) - Query(a - 1) << '\n';
break;
case 2:
in >> a;
out << SearchSum(a) << '\n';
break;
}
}
}