[백준 1417] 국회의원 선거 (Java) - 그리디, 우선순위 큐

https://www.acmicpc.net/problem/1417

국회의원 선거

문제

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

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

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

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

예를 들어서, 마음을 읽은 결과 기호 1번이 5표, 기호 2번이 7표, 기호 3번이 7표 라고 한다면, 다솜이는 2번 후보를 찍으려고 하던 사람 1명과, 3번 후보를 찍으려고 하던 사람 1명을 돈으로 매수하면, 국회의원에 당선이 된다.

돈으로 매수한 사람은 반드시 다솜이를 찍는다고 가정한다.

다솜이가 매수해야하는 사람의 최솟값을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 후보의 수 N이 주어진다. 둘째 줄부터 차례대로 기호 1번을 찍으려고 하는 사람의 수, 기호 2번을 찍으려고 하는 수, 이렇게 총 N개의 줄에 걸쳐 입력이 들어온다. N은 50보다 작거나 같은 자연수이고, 득표수는 100보다 작거나 같은 자연수이다.

출력

첫째 줄에 다솜이가 매수해야 하는 사람의 최솟값을 출력한다.

예제 입력 1 복사

3
5
7
7

예제 출력 1 복사

2

예제 입력 2 복사

4
10
10
10
10

예제 출력 2 복사

1

예제 입력 3 복사

1
1

예제 출력 3 복사

0

예제 입력 4 복사

5
5
10
7
3
8

예제 출력 4 복사

4

---

풀이

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;

public class Main {
    public static void main(String[] args)throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine()); // 후보 수

        // 단일후보라면 매수안해도 되므로 0출력 후 종료
        if (N==1){
            System.out.println(0);
            return;
        }

        int dasom = Integer.parseInt(br.readLine()); // 다솜이 득표 수

        // 내림차순 우선순위 큐
        PriorityQueue<Integer> queue = new PriorityQueue<>((a,b)-> b-a);

        // 다른사람 득표 수 큐에 넣기
        for(int i=1; i<N; i++){
            queue.add(Integer.parseInt(br.readLine()));
        }

        // 매수해야 하는 인원 수
        int count = 0;

        while (!queue.isEmpty()){
            int others = queue.poll(); // 다른후보 득표 수
            if (others >= dasom){ // 다솜이 득표보다 크거나 같다면
                dasom++; // 다솜 득표수 1표 올리고
                count++; // 매수인원 증가
                queue.add(others-1); // 다른후보 득표수 1감소 후 다시 큐에 넣기
            }
        }

        System.out.println(count);
    }
}

 

• 그리디, 우선순위 큐를 활용한 문제
• 다솜이의 득표수가 다른 후보들의 득표수보다만 많으면 국회의원으로 선출되는데, 이때 매수해야 할 사람들의 최소인원 수를 구하는 문제

// 단일후보라면 매수안해도 되므로 0출력 후 종료
        if (N==1){
            System.out.println(0);
            return;
        }

• 만약 단일 후보 (N==1)이면 아무도 매수하지 않아도 되므로 바로 0 출력 후 함수 종료한다.

int dasom = Integer.parseInt(br.readLine()); // 다솜이 득표 수

        // 내림차순 우선순위 큐
        PriorityQueue<Integer> queue = new PriorityQueue<>((a,b)-> b-a);

        // 다른사람 득표 수 큐에 넣기
        for(int i=1; i<N; i++){
            queue.add(Integer.parseInt(br.readLine()));
        }

• N>2일 경우에는 다른 후보를 뽑을 사람들을 매수해야 한다.
  • dasom : 다솜이의 득표 수를 저장한다.
  • 내림차순 정렬 우선순위 큐를 선언한다.
  • 다른후보의 득표 수를 큐에 넣는다.

        // 매수해야 하는 인원 수
        int count = 0;

        while (!queue.isEmpty()){
            int others = queue.poll(); // 다른후보 득표 수
            if (others >= dasom){ // 다솜이 득표보다 크거나 같다면
                dasom++; // 다솜 득표수 1표 올리고
                count++; // 매수인원 증가
                queue.add(others-1); // 다른후보 득표수 1감소 후 다시 큐에 넣기
            }
        }

• 큐가 빌때까지 아래 동작을 반복수행한다.
  • 큐에서 poll 하여 가장 득표수가 많은 수를 확인한다.(others)
  • 만약 others가 dasom 보다 크거나 같을 경우에는 매수해야 한다.
  • dasom의 득표수를 1증가 시키고, count(매수 인원)를 1증가 시킨다.
  • 그리고 큐에 others-1 값을 다시 큐에 넣는다.
  • 만약 others가 dasom 보다 작을 경우에는 다시 큐에 넣을 이유가 없으므로 버린다.
• while문이 끝나면 count 출력 후 함수를 종료시킨다.