Cod sursa(job #936288)

Utilizator vld7Campeanu Vlad vld7 Data 6 aprilie 2013 16:10:25
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <fstream>

using namespace std;

ifstream f("datorii.in");
ofstream g("datorii.out");

const int MAX_N = 15005;

int n, m, a[MAX_N], aint[4 * MAX_N];

void build (int node, int left, int right) {
	if (left == right) {
		aint[node] = a[left];
		return ;
	}
	
	int mid = (left + right) / 2;
	build (2 * node, left, mid);
	build (2 * node + 1, mid + 1, right);
	
	aint[node] = aint[2 * node] + aint[2 * node + 1];
}

void update (int nod, int left, int right, int pos, int val) {
	if (left == right) {
		aint[nod] -= val;
		return ;
	}
	
	int mid = (left + right) / 2;
	if (pos <= mid)
		update (2 * nod, left, mid, pos, val);
	else
		update (2 * nod + 1, mid + 1, right, pos, val);
	
	aint[nod] = aint[2 * nod] + aint[2 * nod + 1];
}

int query (int nod, int left, int right, int a, int b) {
	if (a <= left && right <= b)
		return aint[nod];
	
	int mid = (left + right) / 2;
	int result = 0;
	if (a <= mid)
		result = query (2 * nod, left, mid, a, b);
	if (b > mid)
		result += query (2 * nod + 1, mid + 1, right, a, b);
	
	return result;
}

int main() {
	f >> n >> m;
	for (int i = 1; i <= n; i++)
		f >> a[i];
	
	build (1, 1, n);
	
	while (m--) {
		int op, a, b;
		
		f >> op >> a >> b;
		if (op == 0)
			update (1, 1, n, a, b);		// a[a] -= b
		else
			g << query (1, 1, n, a, b) << '\n';
	}
	
	f.close();
	g.close();
	
	return 0;
}