[BOJ 10800 // C++] 컬러볼

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 10800번 문제인 컬러볼이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/10800 

10800번: 컬러볼

지문에 중요한 조건이 많으니 문제를 주의깊게 읽자.

이 문제는 각 플레이어가 (1)자기 공보다 크기가 작고, (2)자기 공과 색이 다른 공을 사로잡아 점수를 얻을 수 있을 때 각 플레이어마다 얻을 수 있는 최대점수를 구하는 문제이다.

 

자신과 다른 색의 공의 크기의 합을 구하는 것은 모든 공의 크기의 합에서 같은 색의 공의 크기의 합을 빼는 것으로 구할 수 있다는 점을 관찰하자.

 

위의 성질을 이용하면, 공을 크기순으로 정렬하고 작은 크기의 공부터 계산해내가면서 전체 누적 합과 공 색깔별 누적 합을 이용하면 문제를 빠르게 해결할 수 있다.

 

크기가 같은 공이 있는 케이스에 유의하자. 크기가 같은 공은 서로를 사로잡을 수 없기 때문에 서로의 크기가 누적합에 영향을 주어서는 안 된다.

 

아래는 제출한 소스코드이다.

#include <iostream>
#include <algorithm>
#include <vector>
#include <stack>
using namespace std;

int ans[200001];
int psum = 0;
int colorpsum[200001];
vector<pair<int, pair<int, int>>> vec;
stack<pair<int, int>> stk;

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

	int N; cin >> N;
	for (int i = 1; i <= N; i++) {
		int c, w; cin >> c >> w;
		vec.emplace_back(make_pair(w, make_pair(c, i)));
	}

	sort(vec.begin(), vec.end());

	int old = -1;
	for (auto p : vec) {
		if (old < p.first) {
			old = p.first;
			while (!stk.empty()) {
				psum += stk.top().first;
				colorpsum[stk.top().second] += stk.top().first;
				stk.pop();
			}
		}
		ans[p.second.second] = psum - colorpsum[p.second.first];
		stk.push(make_pair(p.first, p.second.first));
	}

	for (int i = 1; i <= N; i++) cout << ans[i] << '\n';
}