Pagini recente » Cod sursa (job #1722984) | Cod sursa (job #1973664) | Cod sursa (job #2899707) | Cod sursa (job #909095) | Cod sursa (job #880883)
Cod sursa(job #880883)
#include <fstream>
using namespace std;
ifstream in("aib.in");
ofstream out("aib.out");
struct AIB {
AIB(int);
~AIB() { delete[] c; }
void add(int, int);
int sum(int, int);
private:
int sumTo(int); // Suma din interv [1, pos]
int * c;
int N;
};
AIB::AIB(int n)
{
c = new int[n+1]; N = n;
for (int i = 1; i <= N; i++) c[i] = 0;
}
void AIB::add(int ind, int val)
{
int poz = 0;
while (ind <= N)
{
c[ind] += val;
while ( (ind & (1 << poz)) == 0 ) poz += 1;
ind += (1 << poz);
poz += 1;
}
}
int AIB::sumTo(int ind)
{
int sum = 0, poz = 0;
while (ind > 0)
{
sum += c[ind];
while ( (ind & (1 << poz)) == 0 ) poz += 1;
ind -= (1 << poz);
poz += 1;
}
return sum;
}
int AIB::sum(int a, int b)
{
return sumTo(b) - sumTo(a-1);
}
int main()
{
int n, m, x, y, op;
in >> n >> m;
AIB bin(n);
for (int i = 1; i <= n; i++)
{
in >> x;
bin.add(i, x);
}
for (int i = 1; i <= m; i++)
{
in >> op;
switch (op)
{
case 0: { in >> x >> y; bin.add(x, y); } break;
case 1: { in >> x >> y; out << bin.sum(x, y) << '\n'; } break;
case 2: {
int k = -1; in >> x;
for (int i = 1; i <= n; i = (i << 1))
{
int s = bin.sum(1, i);
if (s == x) { k = i; break; }
else if (x < s) {
for (int j = (i >> 1) + 1; j < i; j++)
if (s == bin.sum(1, i)) { k = j; goto end; }
}
}
end:
out << k << '\n';
} break;
}
}
return 0;
}