[백준] 22252번 - 정보상인호석 (java)

Baekjoon 22252 - 정보 상인 호석 (클릭 시 이동)

문제

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

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

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

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

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

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

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

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
2 초	512 MB	686	317	217	41.892%

입력

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

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

출력

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

풀이 (Map)

name을 Key로 설정하여 정보들을 value (Collection)에 저장한다.

가장 큰 값순으로 b개만큼 구매하기 때문에 value를 PriorityQueue로 지정했다.

가격이 10만인 정보를 10만 x 5만 = 5,000,000,000원 만큼 구매하게 될 수 있으므로 정답을 Long으로 한다.

최종 소스코드

import java.io.*;
import java.util.*;

public class BJ_G5_22252_정보_상인_호석 {
    static int Q, k, b;
    static long totalCost = 0L;
    static String name;
    static Map<String, PriorityQueue<Integer>> info = new HashMap<>();

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;

        Q = Integer.parseInt(br.readLine());
        for (int i = 0; i < Q; i++) {
            st = new StringTokenizer(br.readLine());
            if (Integer.parseInt(st.nextToken()) == 1) {    // 1일 경우
                name = st.nextToken();
                k = Integer.parseInt(st.nextToken());

                if (!info.containsKey(name)) {        // name이 map에 존재하지 않을 경우 pq를 생성해준다.
                    info.put(name, new PriorityQueue<>(Collections.reverseOrder()));
                }

                PriorityQueue<Integer> pq = info.get(name);
                for (int j = 0; j < k; j++) {
                    pq.offer(Integer.parseInt(st.nextToken()));
                }

            } else {                                        // 2일 경우
                name = st.nextToken();
                if (!info.containsKey(name))                // 없는 정보일 경우 건너뛴다.
                    continue;

                b = Integer.parseInt(st.nextToken());

                PriorityQueue<Integer> pq = info.get(name);

                for (int j = 0; j < b && !pq.isEmpty(); j++) {
                    totalCost += pq.poll();
                }
            }
        }
        System.out.println(totalCost);
    }
}