[C/C++]백준 2042

 

백준 2042 - 구간 합 구하기 (골드1)

20920 문제 링크

세그먼트 트리의 기본 코드를 사용하면 쉽게 풀리는 문제이다.
#include <iostream>
#include <algorithm>
#include <vector>
#define FIO cin.tie(0);cout.tie(0);ios::sync_with_stdio(0);
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#define INF 987654321
using namespace std;
typedef long long ll;
ll n, m, k, h = 1;
vector <ll> tree, num;

void update(int index, ll value) {
	index += h;
	tree[index] = value;
	while (index > 1) {
		index /= 2;
		tree[index] = tree[index * 2] + tree[index * 2 + 1];
	}
}
ll query(int l, int r, int node, int nodel, int noder) {
	if (l <= nodel && noder <= r) {
		return tree[node];
	}
	if (noder < l || r < nodel) {
		return 0;
	}
	int mid = (nodel + noder) / 2;
	return query(l, r, node * 2, nodel, mid) + query(l, r, node * 2 + 1, mid + 1, noder);
}
int main() {
	FIO;
	cin >> n >> m >> k;
	//n: 수의 개수, m: 수의 변경이 일어나는 횟수, k: 구간의 합을 구하는 횟수
	//크기 설정
	while (n > h) {
		h <<= 1;
		tree.resize(h * 2);
		num.resize(n + 1);
	}
	//숫자 넣어주기
	for (int i = 0; i < n; i++) {
		cin >> num[i];
		update(i, num[i]);
	}
	for (int i = 0; i < m + k; i++) {
		ll a, b, c;
		cin >> a >> b >> c;
		//1이면 update해주기
		if (a == 1) {
			update(b - 1, c);
		}
		//2이면 구간의 합 구하기
		else {
			cout << query(b, c, 1, 1, h) << "\n";
		}
	}
}