5-pu2rile

[pu2rile]

🔗 문제 링크

https://www.acmicpc.net/problem/1417

✔️ 소요된 시간

30분

✨ 수도 코드

• 문제 요약

  • 국회의원 후보의 수는 N
  • 다솜이는 기호 1번
  • 다솜이의 득표수가 가장 많도록 다른 후보자의 표를 매수
  • 매수한 표의 최솟값은?

• 알고리즘

  1. 후보의 수 입력(N) -> 총 몇 명이 출마하는지
  2. 다솜이가 기호 1번이므로 다솜이부터 입력
  3. 다솜이를 제외하고 총 몇명이 출마하는지 모르니 리스트 생성 -> vote[]
  4. 다솜이를 제외한 N-1명의 후보자의 득표수를 vote[]에 삽입
  5. 득표수가 많은 사람부터 정렬하기 위해 vote.sort(reverse = True)를 사용하여 내림차순 정렬
  6. 만약 단일 후보라면 (N==1) '0' 출력
  7. 단일 후보가 아니라면 득표수가 가장 많은 후보 vote[0]가 다솜이보다 많거나 같을 때까지
     다솜이 표 + 1 다득표자 - 1 매수해야 하는 사람 수(cnt) +1 => 다시 vote[] 정렬
  8. cnt 출력

💻 코드

N = int(input())
dasom = int(input())
vote = []
cnt = 0

for _ in range(N-1):
  vote.append(int(input()))

vote.sort(reverse=True) #내림차순 정렬

if N == 1:
  print('0')
else:
    while vote[0] >= dasom:
        dasom += 1
        vote[0] -= 1
        cnt += 1
        vote.sort(reverse=True)
    print(cnt)

📚 새롭게 알게된 내용

마침 PR 날이 총선 날이라 국회의원 선거 문제를 풀어 봤습니다!!
출마자 리스트를 어떻게 정리해서 조건을 뽑아낼까 고민을 많이 했는데 .sort(reverse=True)라는 게 있더라구요!!
덕분에 제법 효율적인 코드가 나온 것 같습니닷

[Ghdrn1399]

선거날 국회의원 선거 문제 풀어보는 센스 배우고 갑니다^^
깔끔하게 코드 적어주셔서 이해하기 정말 쉬웠던것 같아요!!

위에서 .sort(reverse=True 언급 해 주셨는데 관련블로그 링크 참고하면 도움 되실것 같아서 링크 걸어두고 갑니다~ㅎㅎ
https://blog.naver.com/wideeyed/221745432175

5번째 PR 수고많으셨어요!!

[Ghdrn1399]

오... 저도 sort(reverse=True)처음보는데 새롭게 알아가요..!!^^

[oesnuj]

총선날이여서 국회의선 선거 문제를 푸셨군요 ㅎㅎ
내림차순 정렬로 현재 가장 많은 표를 가진 후보자를 찾아내는 원리네요
코드 잘 봤습니다!! 이번 PR도 수고하셨습니다👍

[suhyun113]

늦게 리뷰 달아서 죄송합니다😭😭
선거날이라 선거 관련 알고리즘 문제라니,,, 센스 대박이에요!
선거를 조작하는 문제라니..신기하네요ㅎㅎ 문제 잘 봤어요!

이 문제에서는 N명의 후보자 중 다솜이의 표를 다른 후보자들보다 더 많이 받게 하기 위해 필요한 최소한의 표를 증가시키는 횟수를 계산해야 하는데, 도현님 코드를 보니, 직관적으로 잘 이해가 되었어요!

따로 찾아보니, 이 알고리즘 문제는 힙을 이용해서도 풀 수 있더라구요!
도현님이 코드는 가장 내림차순 정렬을 통해 가장 득표수가 많은 사람을 계속 확인해 매수하여 다솜이의 득표수를 올리는 것을 반복하잖아요? 그러다가 제일 득표수가 많은 사람이 다솜이보다 득표수가 낮아지는 순간 멈추는 것을 while문으로 구현했어요. 따라서 가장 득표수가 많은 사람을 알고 싶은 것인데, 매번 정렬하는 것보다는 최대, 최소를 계속 알아야 하는 이런 문제에 좀 더 적합한 자료구조로 힙이 있어요!

힙(우선순위 큐)을 이용하면, 다솜이의 처음 득표수를 저장해 두었다가 다솜이 외의 다른 사람들의 득표수를 힙에 push해 두는 것이에요. 그 후, 똑같이 다솜이의 득표수가 1등이 될 때까지 힙에서 가장 많은 득표수를 가진 사람을 뽑아 그 사람의 표를 매수해 다솜이의 득표수를 증가시켜주고, 힙에 저장된 다른 사람의 득표수를 1 감소시킨 후 다시 힙에 push하는 거에요.
결국, 도현님의 방식과 구현 방식이 거의 똑같은 것으로 보이지만, heap을 이용하면 가장 많은 표를 가진 후보를 좀 더 효율적으로 찾고 업데이트할 수 있어요! 따라서 리스트의 정렬 없이도 가장 큰 값을 빠르게 찾을 수 있어 시간 복잡도를 줄일 수도 있다는 장점이 있어요.(힙 -> O(log N), 정렬 -> O(N log N))

🐧참고🐧

-힙 : 완전 이진 트리의 일종으로 이진트리를 사용하는 자료구조 -> 최대 힙과 최소 힙으로 나뉘며, 부모 자식 간의 관계가 대소 관계를 이루고 있다. 이진 트리는 중복값이 허용되지 않지만, 힙은 중복된 값이 허용된다는 특징을 가지고 있으며, 항상 루트 노드가 가장 작거나 가장 크다. 즉, 힙은 최대, 최소값을 빠르게 찾기 위해 고안된 자료구조이다.

다양한 알고리즘을 통해 푸는 것이 도움이 될 것 같아 아래 코드를 첨부해 봐요!! 참고해보시면 좋을 듯 해요

-힙에서 자료를 추가할 때는 heappush, 자료를 삭제할 때는 heappop을 사용하며, 직접 heap을 구현하는 것보다 파이썬에서 제공하는 heapq 모듈을 이용하면 더 간편하고 빠른 구현이 가능하다.

import heapq

N = int(input())

hq = []
for i in range(N):
    vote = int(input())
    if i == 0 : # 기호 1번 다솜이인 경우
        dasom = vote # 다솜이의 득표수
        continue

    heapq.heappush(hq, -vote) # 다솜이를 제외한 다른 사람들의 득표수

cnt = 0 # 비교 횟수
while hq:
    vote = -heapq.heappop(hq) # 힙에 저장된 득표수 차례대로 비교
    if vote < dasom: #(다솜이의 득표수가 1등이 될 때까지)
        break
    dasom += 1
    cnt += 1
    heapq.heappush(hq, -(vote - 1))
    
print(cnt)