[백준] 구간 합 구하기 (2042번)

문제

어떤 N개의 수가 주어져 있다. 그런데 중간에 수의 변경이 빈번히 일어나고 그 중간에 어떤 부분의 합을 구하려 한다. 만약에 1,2,3,4,5 라는 수가 있고, 3번째 수를 6으로 바꾸고 2번째부터 5번째까지 합을 구하라고 한다면 17을 출력하면 되는 것이다. 그리고 그 상태에서 다섯 번째 수를 2로 바꾸고 3번째부터 5번째까지 합을 구하라고 한다면 12가 될 것이다.

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 1,000,000)과 M(1 ≤ M ≤ 10,000), K(1 ≤ K ≤ 10,000) 가 주어진다. M은 수의 변경이 일어나는 회수이고, K는 구간의 합을 구하는 회수이다. 그리고 둘째 줄부터 N+1번째 줄까지 N개의 수가 주어진다. 그리고 N+2번째 줄부터 N+M+K+1번째 줄까지 세 개의 정수 a, b, c가 주어지는데, a가 1인 경우 b번째 수를 c로 바꾸고 a가 2인 경우에는 b번째 수부터 c번째 수까지의 합을 구하여 출력하면 된다.

a가 1인 경우 c는 long long 범위를 넘지 않는다.

출력

첫째 줄부터 K줄에 걸쳐 구한 구간의 합을 출력한다. 단, 정답은 long long 범위를 넘지 않는다.

 

---

세그먼트 트리를 이용해서 푸는 문제이다. 

구간 합을 구하는 건 세그먼트 트리의 기본격인 문제이다. 스스로 풀 수 있도록 연습을 많이 해야겠다.

struct나 class로 트리를 만들 수도 있지만 간단하게 배열로 해결했다.

배열 값을 모두 받은 후 init()을 통해 구간 합을 모두 전처리 한 후, a값에 따라 update 또는 query를 실행한다.

update는 해당 인덱스를 새 값으로 변경하는 것이고 query는 b부터 c까지 구간 합을 출력하는 것이다.

update를 두 가지 방식으로 해결했는데 void와 longlong을 반환하는 함수 두개를 이용한 방법과 void를 통해 끝낸 방법이 있다. 함수 두 개를 이용하는 방법이 정석급이지만 간편하게 void 하나로 해결했다.

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/BaekJoon/2020/%EA%B5%AC%EA%B0%84%20%ED%95%A9%20%EA%B5%AC%ED%95%98%EA%B8%B0.cpp

chosh95/STUDY

C++ 코드

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int N, M, K;
int a, b, c;
long long p[1000001];
vector<long long> range;

long long init(int left, int right, int node) {
	if (left == right) return range[node] = p[left];
	int mid = (left + right) / 2;
	range[node] = init(left, mid, node * 2) + init(mid + 1, right, node * 2 + 1);
	return range[node];
}
long long query(int left, int right, int node, int nodeLeft, int nodeRight) {
	if (nodeRight < left || right < nodeLeft) return 0;
	if (left <= nodeLeft && nodeRight <= right) return range[node];
	int mid = (nodeLeft + nodeRight) / 2;
	return query(left, right, node * 2, nodeLeft, mid) + query(left, right, node * 2 + 1, mid + 1, nodeRight);
}
/*
long long update(int idx, int newValue, int node, int nodeLeft, int nodeRight) {
	if (nodeLeft > idx || idx > nodeRight) return range[node];
	if (nodeLeft == nodeRight) return range[node] = newValue;
	int mid = (nodeLeft + nodeRight) / 2;
	range[node] = update(idx, newValue, node * 2, nodeLeft, mid) + update(idx, newValue, node * 2 + 1, mid + 1, nodeRight);
	return range[node];
}
void update(int idx, int newValue) {
	update(idx, newValue, 1, 0, N - 1);
}*/

void update(int idx, int newValue, int node, int nodeLeft, int nodeRight) {
	if (nodeLeft > idx || idx > nodeRight) return;
	if (nodeLeft == nodeRight) {
		range[node] = newValue;
		return;
	}
	int mid = (nodeLeft + nodeRight) / 2;
	update(idx, newValue, node * 2, nodeLeft, mid);
	update(idx, newValue, node * 2 + 1, mid + 1, nodeRight);
	range[node] = range[node * 2] + range[node * 2 + 1];
}
int main()
{
	cin >> N >> M >> K;
	for (int i = 0; i < N; i++) cin >> p[i];
	range.resize(N * 4);
	init(0, N - 1, 1);
	for (int i = 0; i < M + K; i++) {
		cin >> a >> b >> c;
		if (a == 1) {
			update(b-1, c,1,0,N-1);
		}
		else {
			cout << query(b - 1 , c - 1, 1, 0, N - 1) << "\n";
		}
	}
}