백준 No.1417 국회의원 선거 - JAVA

문제

다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다. 다솜이는 이 기계를 이용해서 2008년 4월 9일 국회의원 선거를 조작하려고 한다.

다솜이의 기계는 각 사람들이 누구를 찍을 지 미리 읽을 수 있다. 어떤 사람이 누구를 찍을 지 정했으면, 반드시 선거 때 그 사람을 찍는다. 

현재 형택구에 나온 국회의원 후보는 N명이다. 다솜이는 이 기계를 이용해서 그 마을의 주민 M명의 마음을 모두 읽었다.

다솜이는 기호 1번이다. 다솜이는 사람들의 마음을 읽어서 자신을 찍지 않으려는 사람을 돈으로 매수해서 국회의원에 당선이 되게 하려고 한다. 다른 모든 사람의 듣표수 보다 많은 득표수를 가질 때, 그 사람이 국회의원에 당선된다.

3
5
7
7

 

 

이번 문제에서 주어진 예제가 다 통과하는데 계속 9퍼에서 틀리는 일이 계속 반복되었었다.....

이유는 단순했다. 다솜이의 표보다 큰 수를 1표만 빼서 그랬던 것이다. 만약 다솜이가 4이고 다솜이보다 큰 표가 8일 경우 다솜이가 1표 더 많으면 1추가하고 끝나는 식이다. 이게 반복문에 들어가서 압도적으로 큰 수일 경우는 통과를 못했던 것 같다. 그래서  QUEUE에 해당 값을 담음

 

풀이

우선순위 큐를 사용했다. 정렬이 필요하고 값을 계속 추가하고 빼고를 반복해야하기 때문에 우선순위큐를 사용하는데

오름차순이 아닌 내림차순이어야 한다. 만약

 

4 8 9 가 주어진다고 보자

4는 다솜

 

8,9는 우선순위 큐에 9,8로 들어갈 것이다.

 

while(!queue.isEmpty()){
		int num = queue.poll();
        if(temp <= num){
        		temp++;
        		count++;
                queue.add(--num);
        }
}

    이런식의 코드라면 temp는 4이고, num은 처음에는 9일 것이다. if문을 통과해서 temp는 5가 되고, count는 0에서 1로 그리고 queue에 9-1 값이 담아야 한다. 여기서 실수 했던 것이 이런 큰 차이가 나는 값이 있을 경우를 생각 안하고 그냥 한 번 순회하는 걸로 끝냈던 것이다. 이렇게 계속 값이 담겨나가면서 temp가 제일 킹왕짱 큰 값이 된다.

import java.util.*;

public class Main {

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

        int N = sc.nextInt();
        PriorityQueue<Integer> queue = new PriorityQueue<>(Comparator.reverseOrder());

        int temp = sc.nextInt();

        for (int i = 1; i < N ; i++) {
            queue.add(sc.nextInt());
        }
        int count = 0;

            while (!queue.isEmpty()){
                int num = queue.poll();
                if(temp <= num ){
                    count++;
                    temp++;
                    queue.add(--num);
                }
            }

        System.out.println(count);
    }
}