[Silver V] 국회의원 선거 - 1417
문제 링크
성능 요약
- 메모리: 33188 KB
- 시간: 44 ms
분류
- 자료 구조
- 그리디 알고리즘
- 구현
- 우선순위 큐
- 시뮬레이션
문제 설명
다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다. 다솜이는 이 기계를 이용해서 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
<내 코드>
import sys
from heapq import heappush, heappop
# 1. N 후보 입력, N만큼 M 반복입력
N = int(sys.stdin.readline())
candidates = []
# 1-1. 다솜입력
dasom = int(sys.stdin.readline())
for _ in range(N - 1):
# 최대힙으로 - 붙임
M = (-int(sys.stdin.readline()))
heappush(candidates, M)
answer = 0
while True:
if len(candidates) == 0:
break
max_votes = (-heappop(candidates))
if dasom <= max_votes:
answer += 1
dasom += 1
max_votes -= 1
heappush(candidates, -max_votes)
else:
break
sys.stdout.write(str(answer))
# # 튜플 방식으로 생각 했을 때
# # 2. maxheap 으로 tuple로 저장
# for _ in range(N):
# M = int(sys.stdin.readline())
# heappush(candidates, (_,M))
#
# print(candidates)
#
# # 3. 1순위 후보가 최대값 인지 검증 후 아니면 다른 1등 에서 1을 빼고, 1순위에 1추가
# for i in range(1, len(candidates)):
# if candidates[0][1] < candidates[i][1]:- 처음에는 다솜이것 까지 합쳐서 tuple 방식으로 풀이를 생각했음
그러나, 튜플에서 꺼내고 넣고 하는 방식이 복잡한데 이것에서 막혔음
과거에 튜플로 우선순위 정해서 풀었던 방식을 다시 한번 살펴보고 고민해야하겠음
- 단순하게 다솜이는 fix해 두고 max_heap 에서 비교하면서 더해주고 뺴주는 방식이 가장 간단함
누가 최댓값이고, 그 최댓값에 대해서 추가적인 연산을 하는 방식은 다음과같이 heap 방식이 최적임
max() 보다!!!
- 물론 정렬이 있긴하지만, 정렬은 입력값이 많을 때 유용할것으로 추정
- 같이 입력을 받는 인자라고 하더라도 별도로 빼보는것을 생각해보자
<모범 사례>
import sys
import heapq
N = int(sys.stdin.readline())
dasom = int(sys.stdin.readline())
other_votes = [int(sys.stdin.readline()) for _ in range(N - 1)]
answer = 0
if N == 1:
print(0)
else:
max_heap = [-votes for votes in other_votes]
heapq.heapify(max_heap)
while max_heap and dasom <= -max_heap[0]:
max_votes = -heapq.heappop(max_heap)
max_votes -= 1
dasom += 1
answer += 1
heapq.heappush(max_heap, -max_votes)
print(answer)- 리스트컨프리헨션을 바로쓰고
- 종료조건을 while 문에 넣는방식이 가독성 향상에 도움이 될 수있음
<보충 학습>
heapify를 사용한 힙 생성 VS heappush를 사용한 힙 생성
heapify : O(n)
heappush : O(n log n) -> 반복문 이용