[백준] 1417 국회의원 선거

🌭 문제 설명

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

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

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

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

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

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

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

---

🍗 제한 사항

---

🎁 입출력 예시

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

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

---

😎 나의 풀이

from queue import PriorityQueue

N = int(input())
P = [0] * N
# P [5,10,7,3,8]

for i in range(N):
  P[i] = int(input())
  
pq = PriorityQueue()

for i in range(1, N): # 다솜이를 제외한 나머지를 우선순위큐에 넣어줌
  pq.put(-P[i]) # [10,7,3,8] 인데 최솟값이 아닌 최댓값을 반환하기 위해 -를 붙임
  
if N == 1:
  print(0)
else:
  count = 0
  while True:
    max_value = -pq.get() # 값을 빼올 최댓값
    if max_value < P[0]:
      break # 다솜이의 표가 최댓값보다 크면 break

    max_value -= 1 # 다솜이의 표가 더 작다면 최댓값 -1 해주고
    P[0] += 1 # 다솜이의 표에 +1 해준다.
    count += 1 # count도 +1 해주고
    
    pq.put(-max_value) # -1한 최댓값도 다시 우선순위 큐에 넣어야 비교할 수 있기 때문에
    # 다시 큐에 넣어줌
  
  print(count) # count 출력하면 정답

1. 우선 순위 큐를 이용해서 풀었다.
2. 다솜이를 제외한 나머지를 우선순위큐에 넣는다
3. 최댓값을 반환하기 위해 -P[i]로 넣어준다.
4. N이 1이면 후보가 한 명이기 때문에 0을 출력
5. N이 1이 아니면 우선순위 큐에서 빼온 최댓값을 max_value에 저장하고
   다솜이의 표가 최댓값보다 크면 반복문을 빠져나온다.
6. 다솜이의 표가 작다면 max_value에 -1을 해주고 다솜이 표에 +1을 해준다. (가장 많은 표를 가진 후보의 표를 매수하는 것이 이득이기 때문)
7. 표를 추가한 횟수를 하는 count도 +1 해주고 -1을 한 max_value도 다시 우선순위 큐에 넣어서 비교를 해줘야하기 때문에 넣어준다.
8. 다솜이의 표가 최댓값보다 크면 반복문이 종료되면서 count한 횟수를 출력한다.

---

🧵 다른 풀이

n = int(input()) # 후보의 수
dasom = int(input()) # 기호 1번을 찍으려고 하는 사람의 수
vote = [] # 다른 후보를 찍으려고 하는 사람의 수
count = 0 # 매수해야하는 사람의 수
for _ in range(n - 1):
  vote.append(int(input()))
vote.sort(reverse=True) # 내림차순 정렬
if n == 1:
  print(0)
else:
  # 표가 제일 많은 후보를 찍으려고 하는사람을 매수
  while vote[0] >= dasom:
    dasom += 1
    vote[0] -= 1
    count += 1
    vote.sort(reverse=True)
  print(count)

1. 내림차순 정렬을 해서 푸는 방식이다. (우선순위 큐와 비슷)
2. 역순으로 vote 배열을 다시 정렬해줘서 비교한다.

---

• 우선순위 큐를 이용한 풀이가 더 직관적인 것 같다.