1417 - 국회의원 선거

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

💡 이 문제를 보고 내가 생각한 것들, 생각한 순서

다솜이의 투표수 입력받기

다른 후보의 투표수 입력받기

다솜이의 투표수보다 다른 후보의 투표수가 많으면

다솜이 +1 해주고, 다른후보-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

if n == 1:
    print(0)

else:
    m.sort(reverse=True)
    while m[0] >= target:
        target += 1
        m[0] -= 1
        buy += 1
        m.sort(reverse=True)
    print(buy)

import sys
# 국회의원후보 N명
n = int(sys.stdin.readline())
# 다솜이의 표
target = int(sys.stdin.readline())
# 다른후보 2명의 표를 각각 입력받고
m = [int(sys.stdin.readline()) for _ in range(1, n)]
# 매수한사람 수
buy = 0
# 후보가 한명이라면 
if n == 1:
    print(0)
# 후보가 한명이 아닐때
else:
    # 후보들을 내림차순으로 정렬해 제일 많은 표를 받은 사람수를 뽑음
    m.sort(reverse=True)
    # 제일 많이 표를 받은수가 다솜이표와 같거나 크다면
    while m[0] >= target:
        #다솜이에게 표를 하나주고, 후보꺼하나 빼고, 매수자수에 1을 더한다.
        target += 1
        m[0] -= 1
        buy += 1
        # 다시 후보자수 내림차순으로 정렬
        m.sort(reverse=True)
    # 매수자수 출력
    print(buy)

💡 문제에서 중요한 부분

후보자가 한명일때는 매수해야하는 사람이 없으므로 결과는 0이다.

예외인경우는 어떻게 해야할지 생각하기

# 후보가 한명이라면 
if n == 1:
    print(0)

 

 참고

https://lazypazy.tistory.com/entry/백준-1417번-국회의원-선거-파이썬

우선순위 큐로 푼경우

https://jinho-study.tistory.com/1024

https://fre2-dom.tistory.com/63