[알고리즘] 백준 1417 (파이썬 풀이)

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

1417번: 국회의원 선거

문제

다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다. 다솜이는 이 기계를 이용해서 2008년 4월 9일 국회의원 선거를 조작하려고 한다.

다솜이의 기계는 각 사람들이 누구를 찍을 지 미리 읽을 수 있다. 어떤 사람이 누구를 찍을 지 정했으면, 반드시 선거때 그 사람을 찍는다.

현재 형택구에 나온 국회의원 후보는 N명이다. 다솜이는 이 기계를 이용해서 그 마을의 주민 M명의 마음을 모두 읽었다.

다솜이는 기호 1번이다. 다솜이는 사람들의 마음을 읽어서 자신을 찍지 않으려는 사람을 돈으로 매수해서 국회의원에 당선이 되게 하려고 한다. 다른 모든 사람의 득표수 보다 많은 득표수를 가질 때, 그 사람이 국회의원에 당선된다.

예를 들어서, 마음을 읽은 결과 기호 1번이 5표, 기호 2번이 7표, 기호 3번이 7표 라고 한다면, 다솜이는 2번 후보를 찍으려고 하던 사람 1명과, 3번 후보를 찍으려고 하던 사람 1명을 돈으로 매수하면, 국회의원에 당선이 된다.

돈으로 매수한 사람은 반드시 다솜이를 찍는다고 가정한다.

다솜이가 매수해야하는 사람의 최솟값을 출력하는 프로그램을 작성하시오.

입력

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

출력

첫째 줄에 다솜이가 매수해야 하는 사람의 최솟값을 출력한다.

 

해결코드

n = int(input())
li = []
for i in range(n):
    li.append(int(input()))
li2 = li[1:len(li)]
if(len(li) == 1):
    print(0)
else:
    cnt = 0
    for i in range(n-1):
        li2 = sorted(li2, reverse=True)
        while(li[0] <= li2[i]):
            li[0] += 1
            li2[i] -= 1
            cnt += 1
            li2 = sorted(li2, reverse=True)
    print(cnt)

 

문제풀이

일단 이 문제는 우선순위 큐를 이용해야 한다.

=> 다솜이를 뽑는 사람들과 / 나머지 후보를 뽑는 사람들을 구분해야 한다.

후자의 경우 내림차순 정렬을 하여 큰 수가 인덱스 앞쪽으로 오게 하여 전자와 비교해야 한다.

일단 후보의 수 n을 받아준다.

그리고 각 후보를 지지하는 사람들의 수를 담아줄 리스트(li)를 선언해준다.

for문을 이용하여 li에 값들을 입력받아준다.

li2는 다솜이를 제외한 나머지 후보들을 지지하는 사람들의 수를 저장한 리스트이다.

 

만약 li의 크기가 1이라면 비교할 수가 없으므로 0을 출력한다.

그게 아니라면,

cnt값을 0으로 초기화하고

n-1만큼 for문을 돌린다.(n-1만큼인 이유는 뒤에 나올 li2의 길이가 다솜이를 뺀 값이기 때문)

처음에 먼저 li2를 내림차순으로 정렬해준다.(뒤에도 내림차순으로 정렬해주는데 처음에도 선언하는 이유는

li2의 초기 0번째 값이 가장 큰 값이 아닐 수도 있기 때문)

그리고 while문 안에서 다솜이와 나머지 후보들을 비교해준다.

다솜이 지지하는 사람 수보다 li2[i]후보를 지지하는 사람이 같거나 많다면

다솜이 지지하는 사람을 1명 올려주고, li2[i]후보는 1명 내려준다. 그리고 cnt를 1 올려준다.

값이 변경되었으니 li2를 다시 내림차순으로 정렬해준다.

이런 식으로 반복하다 보면 다솜이를 지지하는 사람들이 제일 많아질 것이다.

print(cnt)로 결과값을 출력한다.