본문 바로가기
알고리즘/자료구조

[백준][Python] 1417번 국회의원 선거

by 임짠짠 2023. 2. 1.
반응형
 

1417번: 국회의원 선거

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

www.acmicpc.net

 

코드

import heapq
n = int(input())
vote = []
one = int(input())
for _ in range(n-1):
	heapq.heappush(vote,-int(input()))
ans = 0
while vote:
	first = -heapq.heappop(vote)
	if first < one:
		break
	first -= 1
	heapq.heappush(vote,-first)
	one += 1
	ans += 1
print(ans)

 

 

반응형

댓글