글 읽기 - 어디가 틀린지 잘 모르겠습니다.

skyjpower 5년 전 0

결과는 '틀렸습니다' 입니다.

도움 부탁드립니다.

#include <iostream>
#include <math.h>
#include <vector>

using namespace std;

vector<long long> dt;
vector<long long> tr;

// 초기화
long long init(int node, int begin, int end) {
	if (begin == end) {
		return tr[node] = dt[begin];
	}
	else {
		int mid = (begin + end) / 2;
		return tr[node] = init(node * 2, begin, mid) + init(node * 2 + 1, mid + 1, end);
	}
}

// 구간 합
long long sum(int node, int begin, int end, int left, int right) {
	// 범위 안에 있지 않은 경우
	if (left > end || right < begin) return 0;
	// left ~ right범위 안에 begin~end가 완전 포함인 경우
	if (left <= begin && end <= right) return tr[node];
	// 범위에 걸쳐 있는 경우
	int mid = (begin + end) / 2;
	return sum(node * 2, begin, mid, left, right) + sum(node * 2 + 1, mid + 1, end, left, right);
}

// 수정
void update(int node, int begin, int end, int index, long long diff) {
	if (index < begin || end < index) return;

	tr[node] += diff;
	if (begin != end) {
		int mid = (begin + end) / 2;
		update(node * 2, begin, mid, index, diff);
		update(node * 2 + 1, mid + 1, end, index, diff);
	}
}

int main() {
	ios_base::sync_with_stdio(0);
	//cin.tie(0);

	int n, m, k;
	cin >> n >> m >> k;
    // 트리 사이즈 계산
	int h = ((int)ceil(log2(n))) + 1;

	dt.resize(n);
	tr.resize(1 << h);
    
    // 배열에 데이터 입력
	for (int i = 0; i < n; i++) {
		long long x; cin >> x;
		dt[i] = x;
	}
    // 초기화
	init(1, 0, n - 1);

	int num = m + k;
	while (num--) {
		int iCmd; cin >> iCmd;
		// 변경
		if (iCmd == 1) {
			int idx; long long temp;
			cin >> idx;
			cin >> temp;
			long long diff = temp - dt[idx - 1];
			update(1, 0, n-1, idx-1, diff);
		}
		// 합 구하기
		else if (iCmd == 2) {
			int left, right;
			cin >> left >> right;
			cout << sum(1, 0, n - 1, left-1, right-1) << '\n';
		}
	}

	return 0;
}

skyjpower 5년 전 0

수정 시에 데이터가 담겨져 있는 dt vector의 값을 변경해주지 않았네요.

너무 조급했던 것 같습니다.

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

문제

문제

출처

ICPC

어디가 틀린지 잘 모르겠습니다.

skyjpower 5년 전 0

skyjpower 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말