인프런 커뮤니티 질문&답변

안녕하세요! 아래와 같이 구현하였는데, 주어진 케이스만으로는 검증이 되지 않아서요! 혹시 다른 케이스 제공 부탁드려도 될까요?

- 구현 -

1. ArrayList의 time 부분을 해시맵으로 구현 -> time 종류의 수를 통해 그 수만큼만 반복하도록 while문을 구현하였습니다.(테스트 케이스의 경우 1, 2, 3 으로 3종류이므로 3번 반복)

2. 0번째, 즉 ArrayList의 맨 앞의 부분을 반복 이전에 미리 빼두었습니다. 

3. while 내에 for문을 통해 i를 갱신하며 이전에 탐색했던 부분은 지나가도록 구현하였습니다.

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


class Pay {
    int money;
    int time;

    public Pay(int money, int time) {
        this.money = money;
        this.time = time;
    }
}
public class Main {

    public static void main(String[] args) throws IOException {
        Main T = new Main();
        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt();
        HashMap<Integer, Integer> hm = new HashMap<>();
        ArrayList<Pay> arr = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            int m = sc.nextInt();
            int t = sc.nextInt();
            hm.put(t, hm.getOrDefault(hm.get(t), 0)+1);

            arr.add(new Pay(m, t));
        }

        Collections.sort(arr, new Comparator<Pay>() {
            @Override
            public int compare(Pay o1, Pay o2) {
                if(o2.time == o1.time) {
                    return o2.money - o1.money;
                } else
                    return o2.time - o1.time;
            }
        });

        PriorityQueue<Integer> pQ = new PriorityQueue<>(Collections.reverseOrder());

        int day = arr.get(0).time;
        int key = hm.size();
        int j = 1;
        int result = 0;
        pQ.offer(arr.get(0).money);
        while(key > 0) {
            for(int i=j; i<arr.size(); i++) {
                if(day == arr.get(i).time) {
                    pQ.offer(arr.get(i).money);
                } else {
                    day=arr.get(i).time;
                    j = i;
                    break;
                }
            }
            key --;
            result += pQ.poll();
        }
        System.out.println(result);
    }
}