글 읽기 - 5퍼에서 틀립니다, 반례 모두 다 통과하네요.

swrho 1년 전 1

정답 코드와 비교해봐도 다른게 없는데 잘 모르겠네요.

배열 크기 문제는 아닌것 같습니다.

세그먼트 트리도 구현 잘한것 같습니다.

왜 안되는 것일까요?

#include <iostream>
#include <algorithm>

using namespace std;

int N, M, K;
long long arr[1000005], segTree[4000005];
long long init(int start, int end, int index) {
	if (start == end) { // leaf node
		segTree[index] = arr[start];
		return segTree[index];
	}
	int mid = (start + end) / 2;
	segTree[index] = init(start, mid, index * 2)
		+ init(mid + 1, end, index * 2 + 1);
	return segTree[index];
}
long long sum(int L, int R, int index, int nodeL, int nodeR) {
	if (nodeR < L || R < nodeL) return 0; // 완전히 포함x
	if (L <= nodeL && nodeR <= R) return segTree[index];
	int mid = (nodeL + nodeR) / 2;
	return sum(L, R, index * 2, nodeL, mid)
		+ sum(L, R, index * 2 + 1, mid + 1, nodeR);
}
void update(int start, int end, int index, int target, long long value) {
	if (target < start || end < target) return;
	segTree[index] += value;
	if (start == end) return;
	int mid = (start + end) / 2;
	update(start, mid, index * 2, target, value);
	update(mid + 1, end, index * 2 + 1, target, value);
}
int main() {
	ios_base::sync_with_stdio(0); cin.tie(0);
	//freopen("input.txt", "r", stdin);
	cin >> N >> M >> K;
	for (int i = 0; i < N; i++)
		cin >> arr[i];
	
	init(0, N - 1, 1);

	for (int i = 0; i < M + K; i++) {
		int a, b, c;
		cin >> a >> b >> c;
		if (a == 1) {
			long long temp = arr[b - 1];
			arr[b - 1] = c;
			update(0, N - 1, 1, b - 1, c - temp);
		}
		else {
			cout << sum(b - 1, c - 1, 1, 0, N - 1) << "\n";
		}
	}
}

zenith82114 1년 전 1

43줄에 있는 c가 long long이어야 합니다.

amsminn 1년 전 4

@jihwan0319 루트가 0일때는 L = node * 2 + 1, R = node * 2 + 2로 처리할 수 있습니다

댓글 쓰기

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

5퍼에서 틀립니다, 반례 모두 다 통과하네요.

swrho 1년 전 1

zenith82114 1년 전 1

amsminn 1년 전 4

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말