[BOJ] 1417 국회의원 선거 - JAVA

1. 문제

1417번: 국회의원 선거

다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다. 다솜이는 이 기계를 이용해서 2008년 4월 9일 국회의원 선거를 조작하려고 한다. 다솜이의 기계는 각 사람들이 누구를 찍을지 미리 읽을 수 있다. 어떤 사람이 누구를 찍을지 정했으면, 반드시 선거 때 그 사람을 찍는다.

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

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

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

---

2. 풀이

힙(PriorityQueue)을 이용했다. 매번 정렬하지 않아도 알아서 우선순위로 정렬이 되어서 선택했다. 대신 내림차순 정렬이 되도록 11번째 줄에 람다식을 같이 이용했다.

pQueue가 비어있지 않으면서 다솜이의 숫자가 pQueue.peek(가장 큰 수) 보다 작거나 같을 때 계속 반복한다.

비교하면서 다솜이의 숫자도 +1해준다.

맨 앞의 숫자를 뽑아서 -1 한 값을 다시 pQueue에 넣는 것을 반복하고, 반복할 때마다 카운트를 1 증가시킨다.

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

public class BOJ_1417_국회의원선거 {
   public static void main(String[] args) throws IOException {
      BufferedReader in = new BufferedReader(new InputStreamReader(System.in));

      int N = Integer.parseInt(in.readLine());
      int dasom = Integer.parseInt(in.readLine());

      PriorityQueue<Integer> pQueue = new PriorityQueue<>((o1,o2)->o2.compareTo(o1));

      for(int i=0;i<N-1;i++) pQueue.offer(Integer.parseInt(in.readLine()));

      int cnt = 0;
      while(!pQueue.isEmpty() && dasom++<=pQueue.peek()){
         pQueue.offer(pQueue.poll()-1);
         ++cnt;
      }

      System.out.println(cnt);

   }
}

---

3. 결과