[우선순위 큐] 백준 1417번 국회의원 선거(Java)

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

1417번: 국회의원 선거

📚 문제

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

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

 

예제 입력	예제 출력
3 5 7 7	2
4 10 10 10 10	1
1 1	0
5 5 10 7 3 8	4

 

📝 문제 해결

우선순위 큐에 다른 후보자들의 득표수를 넣어준 다음

우선순위 큐가 비어있지 않으면서 다솜이의 득표수가 큐의 가장 위의 원소(가장 큰 수)보다 작거나 같을 때,

1. 우선순위 큐 가장 위의 득표수 -1 해주고 다시 우선순위 큐에 삽입
2. 다솜이 득표수에 +1 
3. 다솜이의 득표수가 우선순위 큐의 가장 위의 원소보다 클때까지 반복

 

💻 코드

import java.io.*;
import java.util.*;
 
public class BOJ1417{
   public static void main(String[] args) throws IOException {
      BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
      int N = Integer.parseInt(br.readLine());
      int dasom = Integer.parseInt(br.readLine());
 
      PriorityQueue<Integer> queue = new PriorityQueue<>(Collections.reverseOrder());
 
      for(int i=1; i<N; i++) queue.offer(Integer.parseInt(br.readLine()));
 
      int cnt = 0;
      while(!queue.isEmpty() && dasom <= queue.peek()){
         queue.offer(queue.poll()-1); 	// 우선순위 큐 가장 위의 득표수 -1
         dasom++;	// 다솜이 득표수 +1 
         cnt++;
      }
 
      System.out.println(cnt);
 
   }
}