백준 2042 - 구간 합 구하기 (골드1)
세그먼트 트리의 기본 코드를 사용하면 쉽게 풀리는 문제이다.
#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";
}
}
}
PREVIOUS머신러닝-지도학습알고리즘2