[baekjoon] 백준 1417번(파이썬): 국회의원 선거

문제

1417번: 국회의원 선거

- 다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다.

- 다솜이는 이 기계를 이용해서 2008년 4월 9일 국회의원 선거를 조작하려고 한다.

- 다솜이의 기계는 각 사람들이 누구를 찍을지 미리 읽을 수 있다.

- 어떤 사람이 누구를 찍을지 정했으면, 반드시 선거 때 그 사람을 찍는다.

- 현재 형택구에 나온 국회의원 후보는 N명이다.

- 다솜이는 이 기계를 이용해서 그 마을의 주민 M명의 마음을 모두 읽었다.

- 다솜이는 기호 1번이다.

- 다솜이는 사람들의 마음을 읽어서 자신을 찍지 않으려는 사람을 돈으로 매수해서 국회의원에 당선이 되게 하려고 한다.

- 다른 모든 사람의 득표수 보다 많은 득표수를 가질 때, 그 사람이 국회의원에 당선된다.

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

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

- 다솜이가 매수해야하는 사람의 최솟값을 구하는 문제이다.

- 후보의 수 N이 주어진다

- 다음은 차례대로 기호 1번을 찍으려고 하는 사람의 수, 기호 2번을 찍으려고 하는 수, 이렇게 총 N개의 줄에 걸쳐 입력이 들어온다.

- N은 1,000보다 작거나 같은 자연수이고, 득표수는 1,000보다 작거나 같은 자연수이다.

알고리즘

- 후보의 수를 입력받는다.

- 다솜이의 투표수를 입력받고 다른 후보들의 투표수를 입력받는다.

- 후보가 1명이면 매수할 필요가 없으므로 0을 출력한다.

- 다른 후보들의 투표수를 내림차순 하여 큰 투표수와 다솜이의 투표수를 반복하여 비교한다.

- 다솜이의 투표수가 더 작으면 다솜이의 투표수를 +1 하고 제일 큰 투표수를 -1 하고 매수 횟수를 카운트한다.

코드

import sys

n = int(sys.stdin.readline())
# 다솜이의 투표수
target = int(sys.stdin.readline())
# 다른 후보들의 투표수
m = [int(sys.stdin.readline()) for _ in range(1, n)]
# 매수 횟수
buy = 0

# 후보가 1명이면 다솜이가 당선
if n == 1:
    print(0)
else:
    # 다른 후보들의 투표수를 내림차순하여 큰 투표수와 다솜이의 투표수를 비교한다.
    m.sort(reverse=True)
    # 다솜이의 투표수가 제일 클때까지 반복한다.
    # 다솜이의 투표수를 +1 하고 제일 큰 투표수를 -1 한다.
    # 매수 했으므로 매수를 카운트한다.
    while m[0] >= target:
        target += 1
        m[0] -= 1
        buy += 1
        # 다시 다른 후보들의 투표수를 내림차순하여 제일 큰 투표수와 비교한다.
        m.sort(reverse=True)
    print(buy)

github

junjange/CodingTest