[백준] 국회의원 선거 (Python)

문제

다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다. 다솜이는 이 기계를 이용해서 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

예제 입력 2

4
10
10
10
10

예제 출력 2

1

예제 입력 3

1
1

예제 출력 3

0

예제 입력 4

5
5
10
7
3
8

예제 출력 4

4

최종 코드

# 후보는 N명
# 기호 1번, 2번, ,,, N번 한줄씩
# 다솜이는 기호 1번
# 한번 1을 빼오고 비교 하고 다른 후보표를 빼고 비교하고 반복해야함 -- 기호1번이 제일 높은 수면 종료
# 그 반복 값 출력하기
# 가장 큰 수를 찾아서 거기서 빼줘야함
# 1을 뺀 값을 다시 그 후보 수에 넣어서 반복

import sys

# 전체 후보자 수
N = int(sys.stdin.readline())
# 후보자 리스트
vote_list = [int(sys.stdin.readline().strip()) for _ in range(N)]

# 기호1번 다솜쿤 투표 수
dasom_vote = vote_list[0]
# 기호2번부터 다른 경쟁자 투표 수
opponent_vote = vote_list[1:]
# 주작 횟수
count = 0

# while dasom_vote > opponent_vote[max_index]:
#   max_number = max(opponent_vote)
#   max_index = opponent_vote.index(max_number)
#   opponent_vote[max_index] -= 1
#   dasom_vote += 1
#   count += 1
  
# print(count)
# ========================================
# opponent_vote[max_index]값이 while문 내부에서 선언한 값임

# while dasom_vote <= max(opponent_vote):
#   max_number = max(opponent_vote)
#   max_index = opponent_vote.index(max_number)
#   opponent_vote[max_index] -= 1
#   dasom_vote += 1
#   count += 1
# ===============================
# 기호1번만 있는 예제 3의 케이스 처리를 안해줌

# 경쟁자가 있거나, 다솜이 투표가 다른 투표자 수보다 같거나 작을때 계속 while문 반복
while opponent_vote and dasom_vote <= max(opponent_vote):
  # 경쟁자 가장 높은 투표 수
  max_number = max(opponent_vote)
  # 높은 수의 index
  max_index = opponent_vote.index(max_number)
  # 해당 index의 숫자 -1
  opponent_vote[max_index] -= 1
  # 다솜이 투표 수 +1
  dasom_vote += 1
  # 주작 횟수 +1
  count += 1
  
print(count)

풀이 과정

import sys

# 전체 후보자 수
N = int(sys.stdin.readline())
# 후보자 리스트
vote_list = [int(sys.stdin.readline().strip()) for _ in range(N)]

# 기호1번 다솜쿤 투표 수
dasom_vote = vote_list[0]
# 기호2번부터 다른 경쟁자 투표 수
opponent_vote = vote_list[1:]
# 주작 횟수
count = 0

1. 일단 sys 라이브러리로 전체 후보자수는 N으로, 0부터 N까지의 범위 반복하면서 int화 시킨 값들은 후보자 리스트인 vote_list안에 담아준다
2. dasome이는 무조건 기호 1번이므로, [0]번째 값을 박아두고, 나머지 투표자들은 [1:]부터 끝까지 opponent_vote 값에 담아둔다.
3. 주작 횟수는 count로 체크할 예정

# while dasom_vote > opponent_vote[max_index]:
#   max_number = max(opponent_vote)
#   max_index = opponent_vote.index(max_number)
#   opponent_vote[max_index] -= 1
#   dasom_vote += 1
#   count += 1
  
# print(count)
# ========================================
# opponent_vote[max_index]값이 while문 내부에서 선언한 값임

while문으로 다솜이 투표수가 나머지 투표수들보다 커질때까지 가장 큰 투표 숫자를 max_number에, 그리고 그 number가 위치한 index를 max_index로 설정해서 계~속 그 인덱스엔 -1을 다솜이 숫자엔 +1을, 주작횟수를 +1해주는 식으로 구현했는데, 여기서 while 조건문에 opponent_vote[max_index]가 내부 로직에서 선언한 값이기에 다르게 표현해줄 필요가 있었다
단, while문을 자세히 보면, 다솜이가 더 클 경우에 while문을 종료시켜야되는데 반대로 기재하였다.

# while dasom_vote <= max(opponent_vote):
#   max_number = max(opponent_vote)
#   max_index = opponent_vote.index(max_number)
#   opponent_vote[max_index] -= 1
#   dasom_vote += 1
#   count += 1
# ===============================
# 기호1번만 있는 예제 3의 케이스 처리를 안해줌

max 매서드를 활용해서 가장 큰 값이 다솜이 투표수보다 크거나 같을땐 로직을 수행하게 변경하였고, 나머지는 동일하게 코드 구현하였다. 그러나 한가지 망각한건. 예제 3번처럼 다른 투표자가 없는 경우에 로직처리를 진행해주지 않았다.

# 경쟁자가 있거나, 다솜이 투표가 다른 투표자 수보다 같거나 작을때 계속 while문 반복
while opponent_vote and dasom_vote <= max(opponent_vote):
  # 경쟁자 가장 높은 투표 수
  max_number = max(opponent_vote)
  # 높은 수의 index
  max_index = opponent_vote.index(max_number)
  # 해당 index의 숫자 -1
  opponent_vote[max_index] -= 1
  # 다솜이 투표 수 +1
  dasom_vote += 1
  # 주작 횟수 +1
  count += 1
  
print(count)

while문에 opponent_vote가 있을 경우까지 담아줘서 문제 해결!!