백준 22252 정보 상인 호석 c++,Kotlin(해시,우선순위 큐)

문제 출처 : 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일 땐, 고릴라의 이름이 주어지고, 해당 이름을 가진 고릴라가 알고 있는 정보의 가치가 주어진다.

한 고릴라에 대해 여러 번 정보의 가치가 추가될 수 있으니, 

중복을 없애는 map 자료구조를 생각해 볼 수 있다.

또, 고릴라가 가지고 있는 정보의 개수는 주어지지 않았으니
동적으로 할당하기 위해 vector 자료 구조를 생각할 수 있다.

하지만 vector자료구조 사용 시 고릴라가 정보를 얻고, 파는 과정에서 데이터의 삽입, 삭제 과정이 많이 일어날 것이고,

최댓값을 찾기 위해 정렬까지 해줘야 한다. 따라서 데이터의 삽입, 삭제가 빠르고, 정렬까지 해주는 우선순위 큐를 사용하면 간편할 것 같다!

이러한 의식의 흐름으로 풀이를 생각했고, 요즘 c++으로 풀었던 문제들을 코틀린으로 다시 풀고 있는데,

처음 코틀린으로 풀 땐 코틀린에 우선순위 큐가 있는지 모르고 ArrayList로 했다가 시간 초과로 혼났다.

 

코드(c++)

#include <iostream>
#include <queue>
#include <vector>
#include <unordered_map>
#include <string>
using namespace std;
int T;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin >> T;
	unordered_map<string, priority_queue<int>> gorilla;
	long long answer = 0;
	for (int i = 0; i < T; i++) {
		int order;
		string name;
		int n;
		cin >> order;
		cin >> name;
		cin >> n;
		if (order == 1) {
			for (int j = 0; j < n; j++) {
				int price;
				cin >> price;
				gorilla[name].push(price);
			}
		}
		else {
			while (n-- && !gorilla[name].empty()) {
				answer += gorilla[name].top();
				gorilla[name].pop();
			}
		}
	}
	cout << answer << '\n';
	return 0;
}

코드(Kotlin)

import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.*
fun main() = with(System.out.bufferedWriter()) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    var t = Integer.parseInt(br.readLine())
    var answer = 0L
    var gorilla = mutableMapOf<String, PriorityQueue<Int>>()
    while (t-- != 0) {
        val tk = StringTokenizer(br.readLine())
        val order = Integer.parseInt(tk.nextToken())
        val name = tk.nextToken()
        var cnt = Integer.parseInt(tk.nextToken())

        if (order == 1) {
            for (i in 0 until cnt) {
                if (gorilla.get(name) == null) {
                    gorilla.put(name, PriorityQueue<Int>(Collections.reverseOrder()))
                }
                gorilla[name]?.add(Integer.parseInt(tk.nextToken()))
            }
        } else {
            if (gorilla[name]!=null) {
                while (cnt-- != 0 && !gorilla[name]!!.isEmpty()) {
                    answer+=gorilla[name]!!.poll()

                }
            }

        }
    }
    write("$answer")
    close()
    br.close()
}