Cod sursa(job #586341)

Utilizator xaphariusMihai Suteu xapharius Data 30 aprilie 2011 15:58:48
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.11 kb
#include <stdio.h>
#include <stdlib.h>
using namespace std;

#define NMAX 100001


int N, M, aib[NMAX];

int cool(int x){
	return x & -x;
}

void update(int index, int value){
	for (int i = index; i < N + 1; i += cool(i))
		aib[i] += value;
}

int query(int index){
	int sum = 0;
	for (int i = index; i > 0; i -= cool(i))
		sum += aib[i];
	return sum;	
}

int search(int value){
	int step;
	for (step = 1; step < N + 1; step <<= 1);
	for (int i = 0; step >= 0; step >>= 1){
		if (i + step < N + 1 && aib[i + step] <= value){
			i += step;
			value -= aib[i];
			if (!value) return i;
		}
	}
	return -1;
}


int main(){
	freopen("aib.in", "r", stdin);
	freopen("aib.out", "w", stdout);
	
	scanf("%d %d", &N, &M);

	for (int i = 1, x; i < N + 1; ++i){
		scanf("%d", &x);
		update(i, x);
	}
	
	for (int i = 0, cod, a, b; i < M; ++i){
		scanf("%d", &cod);
		
		if(cod == 0){
			scanf("%d %d", &a , &b); 	
			update(a,b);
		}else if (cod == 1){
			scanf("%d %d", &a , &b); 	
			printf("%d\n", query(b) - query(a - 1));
		}
		else {
			scanf("%d", &a);
			printf("%d\n", search(a));
		}
	}


	fclose(stdout);
	fclose(stdin);

	return 0;
}