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

김동욱님의 프로필 이미지
김동욱

작성한 질문수

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

4. 최대수입스케쥴( PriorityQueue)

최대수입스케쥴(PriorityQueue) 질문입니다.

작성

·

3

0

- 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!
- 먼저 유사한 질문이 있었는지 검색해보세요.
- 서로 예의를 지키며 존중하는 문화를 만들어가요.
- 잠깐! 인프런 서비스 운영 관련 문의는 1:1 문의하기를 이용해주세요.

아래와 같이 코드를 작성하였을 때 예제에는 정상정답이 나오고, 저도 반례를 찾지 못하는데, 채점에서는 오답이라고 나옵니다.

어느부분이 오답인지, 그리고 반례가 뭐가 있을지 궁금합니다.ㅠㅠ

 

import java.util.*;



public class Main {

    public static class schedule implements Comparable<schedule>{
        int pay = 0;
        int day = 0;

        schedule(int p,int d) {
            this.pay = p;
            this.day = d;
        }

        @Override
        public int compareTo(schedule o) {
            if(this.day == o.day) return o.pay - this.pay;
            else return o.day - this.day;
        }
    }

    public void solution(List<schedule> list) {
        Collections.sort(list);
        int answer = 0;
        int day = list.get(0).day;
        Queue<Integer> queue = new PriorityQueue<>(Collections.reverseOrder());

        for(schedule s : list) {
            if(day == s.day) {
                queue.offer(s.pay);
            } else {
                if(!queue.isEmpty()) {
                    int p = queue.poll();
                    answer += p;
                }
                queue.offer(s.pay);
                day--;
            }

        }
        System.out.println(answer += queue.poll());

    }


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

        int n = kb.nextInt();
        List<schedule> list = new ArrayList<>();
        for(int i=0; i<n; i++) {
            int pay = kb.nextInt();
            int day = kb.nextInt();
            list.add(new schedule(pay, day));
        }

        T.solution(list);


    }
}

답변

답변을 기다리고 있는 질문이에요
첫번째 답변을 남겨보세요!
김동욱님의 프로필 이미지
김동욱

작성한 질문수

질문하기