Pagini recente » Cod sursa (job #2018470) | Cod sursa (job #736936) | Cod sursa (job #954773) | Cod sursa (job #1773713) | Cod sursa (job #2227057)
#include <stdio.h>
#define MAX_N 100001
int n, m;
int bit[MAX_N];
void print_array(int *v, int a, int b)
{
printf("[");
for (int i = a; i < b; i++)
printf("(%d) %d, ", i, v[i]);
printf("(%d) %d]\n", b, v[b]);
}
void update(int idx, int val)
{
while (idx <= n) {
bit[idx] += val;
idx += (idx & -idx);
}
}
int query(int idx)
{
int sum = 0;
while (idx > 0) {
sum += bit[idx];
idx -= (idx & -idx);
}
return sum;
}
int search(int val)
{
int tidx, idx = 0;
int step = 1;
while (step < n) {
step <<= 1;
}
while (step > 0 && idx < n) {
tidx = idx + step;
if (bit[tidx] <= val) {
idx = tidx;
val -= bit[tidx];
if (val == 0)
return tidx;
}
step >>= 1;
}
return -1;
}
int main(void)
{
// read input
int op, a, b;
freopen("aib.in", "r", stdin);
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &a);
update(i, a);
}
// print output
freopen("aib.out", "w", stdout);
for (int i = 0; i < m; i++) {
scanf("%d", &op);
switch (op) {
case 0: scanf("%d%d", &a, &b);
update(a, b);
break;
case 1: scanf("%d%d", &a, &b);
printf("%d\n", query(b) - query(a - 1));
break;
case 2: scanf("%d", &a);
printf("%d\n", search(a));
break;
}
}
return 0;
}