Cod sursa(job #809308)

Utilizator ahmed.abdraboahmed.abdrabo ahmed.abdrabo Data 8 noiembrie 2012 09:02:41
Problema Arbori indexati binar Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 kb
#include <cstdio>
using namespace std;

const int H = 17;
const int N = 1 << H;
long long BIT[N + 1];

inline long long next_int() {
	long long lld;
	scanf("%lld", &lld);
	return lld;
}

long long get(int index) {
	long long sum = 0;
	while (index) {
		sum += BIT[index];
		index -= index & -index;
	}
	return sum;
}

void add(int index, long long value) {
	while (index <= N) {
		BIT[index] += value;
		index += index & -index;
	}
}

int find(long long sum) {
	int lo = 0, hi = N;
	while (hi - lo > 1) {
		int mid = (lo + hi) / 2;
		if (get(mid) < sum) {
			lo = mid;
		} else {
			hi = mid;
		}
	}
	return get(hi) == sum ? hi : -1;
}

int main() {
	freopen("aib.in", "r", stdin);
	freopen("aib.out", "w", stdout);
	int n = next_int();
	int m = next_int();
	for (int i = 1; i <= n; i++) {
		add(i, next_int());
	}
	for (int i = 0; i < m; i++) {
		long long q = next_int();
		if (q == 0) {
			long long a = next_int();
			long long b = next_int();
			add(a, b);
		}
		if (q == 1) {
			long long a = next_int();
			long long b = next_int();
			printf("%d\n", get(b) - get(a - 1));
		}
		if (q == 2) {
			long long a = next_int();
			printf("%d\n", find(a));
		}
	}
	return 0;
}