forked from ndb796/python-for-coding-test
-
Notifications
You must be signed in to change notification settings - Fork 0
/
3.cpp
62 lines (56 loc) · 1.67 KB
/
3.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#include <bits/stdc++.h>
using namespace std;
// 전체 데이터의 개수는 최대 1,000,000개
long long arr[1000001], tree[1000001];
// 데이터의 개수(n), 변경 횟수(m), 구간 합 계산 횟수(k)
int n, m, k;
// i번째 수까지의 누적 합을 계산하는 함수
long long prefixSum(int i) {
long long result = 0;
while(i > 0) {
result += tree[i];
// 0이 아닌 마지막 비트만큼 빼가면서 이동
i -= (i & -i);
}
return result;
}
// i번째 수를 dif만큼 더하는 함수
void update(int i, long long dif) {
while(i <= n) {
tree[i] += dif;
i += (i & -i);
}
}
// start부터 end까지의 구간 합을 계산하는 함수
long long intervalSum(int start, int end) {
return prefixSum(end) - prefixSum(start - 1);
}
int main(void) {
scanf("%d %d %d", &n, &m, &k);
for(int i = 1; i <= n; i++) {
long long x;
scanf("%lld", &x);
arr[i] = x;
update(i, x);
}
int count = 0;
while(count++ < m + k) {
int op;
scanf("%d", &op);
// 업데이트(update) 연산인 경우
if(op == 1) {
int index;
long long value;
scanf("%d %lld", &index, &value);
update(index, value - arr[index]); // 바뀐 크기(dif)만큼 적용
arr[index] = value; // i번째 수를 value로 업데이트
}
// 구간 합(interval sum) 연산인 경우
else {
int start, end;
scanf("%d %d", &start, &end);
printf("%lld\n", intervalSum(start, end));
}
}
return 0;
}