[BOJ/Gold 5] 백준 22252 정보 상인 호석(C++)

문제 링크

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

22252번: 정보 상인 호석

 

문제

암흑가의 권력은 주먹과 정보에서 나온다. 주먹은 한 명에게 강하고, 정보는 세계를 가지고 놀 수 있기 때문에 호석이는 세상 모든 정보를 모으는 "정보 상인"이 되고 싶다. 정보 상인은 정보를 사고파는 사람을 의미한다.

호석이는 아직 상인계의 새싹이기 때문에, 초기 투자를 통해서 여러 명의 "정보 고릴라"들로부터 정보를 모으려고 한다. 정보 고릴라란 여기저기서 정보를 수집하는 사람들을 의미한다. 일단 정보를 긁어모으기 위해서 호석이는 여러 정보 고릴라들에게 정보를 구매하려고 한다.

암흑가의 연락망은 빼곡하기 때문에 누가 어떤 정보를 얻었는지에 대한 찌라시들이 수시로 퍼진다. 찌라시로 알 수 있는 것은, 어떤 이름을 가진 고릴라가 C1, C2, ..., Ck만큼의 가치가 있는 정보 k 개를 얻었다는 점이다.

호석이는 이를 바탕으로 임의의 시점에 특정 고릴라에게 정보를 몇 개 살 것인지를 정할 수 있다. 이때 가치 순으로 가장 비싼 정보들을 구매한다. 예를 들어 고릴라가 가진 정보가 10개이고, 호석이가 사고 싶은 정보 개수가 4개라면, 고릴라는 10개 중에서 가치 순으로 가장 비싼 4개를 팔 것이다. 한 번 거래한 정보는 호석이에게 더 이상 가치가 없기 때문에 고릴라도 그 정보를 파기한다.

당신은 암흑가의 주먹이며 양대 산맥이 될 가능성이 있는 호석이를 주시하고 있다. 관찰하면서 얻은 정보는 총 Q개이다. 각 정보는 다음의 2가지 중 하나이다.

• 1 Name k C1,C2,...,Ck : 이름이 [Name]인 고릴라가 k개의 정보를 얻었으며, 각 가치는 C1부터 Ck이다.
• 2 Name b : 호석이가 이름이 [Name]인 고릴라에게 b개의 정보를 구매한다. 이때 고릴라가 가진 정보들 중 가장 비싼 b개를 구매하며, 고릴라가 가진 정보가 b개 이하이면 가진 모든 정보를 구매한다.

 견제를 위해서 호석이가 가진 정보들의 가치 총합, 즉 호석이가 정보들을 구매하는 데에 쓴 돈의 총합을 구하자.

입력

고릴라들이 정보를 얻는 사건과 호석이가 거래하는 정보가 시간순으로 주어진다. 첫 번째 줄에는 쿼리의 개수 Q가 주어진다.

이어서 Q개의 줄에 걸쳐서 각 줄에 쿼리가 주어진다. 쿼리는 1이나 2로 시작한다. 1로 시작하는 경우에는 정보를 얻은 정보 고릴라의 이름과 k가 주어지며 이어서 k개의 정보 가치 C1,...,Ck가 자연수로 주어진다. 모든 Ci는 1 이상 100,000 이하이다. 2로 시작하는 경우에는 호석이가 거래하려는 정보 고릴라의 이름과 구매하려는 정보의 개수 b가 주어진다. 

출력

모든 쿼리가 종료되었을 때에 호석이가 얻게 되는 정보 가치의 총합을 출력하라.

제한

• 1 ≤ Q ≤ 100,000, Q는 자연수
• 모든 Name은 알파벳 소문자 혹은 대문자로 이루어져 있고 공백이 없으며 길이는 1 이상 15 이하이다.
• 1 ≤ k ≤ 100,000, k는 자연수
• 1 ≤ C ≤ 100,000, C는 자연수
• 1 ≤ b ≤ 100,000, b는 자연수
• 모든 쿼리에 대한 k의 합은 1,000,000 을 넘지 않는다. 

예제 입력 1

7
1 Cpp 5 10 4 2 8 4
1 Java 2 8 2
2 Cpp 2
1 Cpp 2 10 3
2 Cpp 3
2 Java 1
2 Python 10

예제 출력 1

44

알고리즘 분류

• 자료 구조

풀이

우선 고릴라의 순번을 기록하기 위한 map 자료구조가 하나 필요하고, 각 고릴라가 가지고 있는 정보를 저장하기 위한 우선순위 큐 배열이 필요하다.

쿼리가 1이면 map 자료구조를 이용하여 일단 해당 고릴라의 이름을 key, 번호를 value로 저장하고, 우선순위 큐에 정보들을 저장한다.

쿼리가 2면 우선순위 큐에 있는 정보들 중 가치가 높은 순서대로 B개만큼을 꺼낸다. 만약 B보다 우선순위 큐에 들어있는 원소들의 개수가 더 적다면 모든 정보들을 꺼낸다.

Q개의 쿼리를 처리한 후 호석이가 얻은 정보들의 가치의 총합을 출력한다.

코드

#include <iostream>
#include <string>
#include <cmath>
#include <set>
#include <unordered_map>
#include <bitset>
#include <ctime>
#include <stack>
#include <queue>
#include <vector>
#include <algorithm>
#include <climits>
#define FASTIO cin.tie(NULL); cout.tie(NULL); ios::sync_with_stdio(false);
#define MAX 100001
#define LL long long
#define INF 1e9

using namespace std;
LL Q, N, K, B, C;
LL IDX = 0;
string Name;
unordered_map<string, int> MAP;
priority_queue<LL> PQ[MAX];
LL Answer = 0;

void Input() {
	cin >> Q;
}

void Settings() {
	for (int i = 0; i < Q; i++) {
		cin >> N >> Name;
		if (MAP.find(Name) == MAP.end()) {
			MAP.insert(make_pair(Name, IDX++));
		}
		if (N == 1) {
			cin >> K;
			for (int j = 0; j < K; j++) {
				cin >> C;
				PQ[MAP[Name]].push(C);
			}
		}
		else if (N == 2) {
			cin >> B;
			LL Size = PQ[MAP[Name]].size();
			for (int i = 0; i < min(B, Size); i++) {
				Answer += PQ[MAP[Name]].top();
				PQ[MAP[Name]].pop();
			}
		}
	}
}

void Find_Answer() {
	cout << Answer << "\n";
}

int main() {
	FASTIO
	Input();
	Settings();
	Find_Answer();

	return 0;
}