[lv3] 1713. 후보 추천하기(Python 파이썬 풀이)

출처

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

1713번: 후보 추천하기

결과

문제

월드초등학교 학생회장 후보는 일정 기간 동안 전체 학생의 추천에 의하여 정해진 수만큼 선정된다. 그래서 학교 홈페이지에 추천받은 학생의 사진을 게시할 수 있는 사진틀을 후보의 수만큼 만들었다. 추천받은 학생의 사진을 사진틀에 게시하고 추천받은 횟수를 표시하는 규칙은 다음과 같다.

학생들이 추천을 시작하기 전에 모든 사진틀은 비어있다.
어떤 학생이 특정 학생을 추천하면, 추천받은 학생의 사진이 반드시 사진틀에 게시되어야 한다.
비어있는 사진틀이 없는 경우에는 현재까지 추천 받은 횟수가 가장 적은 학생의 사진을 삭제하고, 그 자리에 새롭게 추천받은 학생의 사진을 게시한다. 이때, 현재까지 추천 받은 횟수가 가장 적은 학생이 두 명 이상일 경우에는 그러한 학생들 중 게시된 지 가장 오래된 사진을 삭제한다.
현재 사진이 게시된 학생이 다른 학생의 추천을 받은 경우에는 추천받은 횟수만 증가시킨다.
사진틀에 게시된 사진이 삭제되는 경우에는 해당 학생이 추천받은 횟수는 0으로 바뀐다.
후보의 수 즉, 사진틀의 개수와 전체 학생의 추천 결과가 추천받은 순서대로 주어졌을 때, 최종 후보가 누구인지 결정하는 프로그램을 작성하시오.

입력

첫째 줄에는 사진틀의 개수 N이 주어진다. (1 ≤ N ≤ 20) 둘째 줄에는 전체 학생의 총 추천 횟수가 주어지고, 셋째 줄에는 추천받은 학생을 나타내는 번호가 빈 칸을 사이에 두고 추천받은 순서대로 주어진다. 총 추천 횟수는 1,000번 이하이며 학생을 나타내는 번호는 1부터 100까지의 자연수이다.

출력

사진틀에 사진이 게재된 최종 후보의 학생 번호를 증가하는 순서대로 출력한다.

예제 입력

사진틀에 사진이 게재된 최종 후보의 학생 번호를 증가하는 순서대로 출력한다.

예제 출력

2 6 7

해석

추천을 받은 후보자를 n개의 액자틀 안에 어떻게 넣을 것이냐는 문제이다. 조건은 아래와 같이 생각해보면 된다.

1. 등록된 후보자 수가 n개 이하인 경우
   • 이미 액자틀에 있는 경우
   • 액자틀에 등록 안되어 있는 경우
2. 등록된 후보자 수가 이미 n개 인 경우
   • 이미 액자틀에 있는 경우
   • 액자틀에 등록 안되어 있는 경우

다른 것들은 쉽게 해결이 가능하나, 2-2는 고려할 조건이 다양하게 된다. 이미 등록된 후보자 수가 n명인데 액자틀에 없는 경우에는 가장 추천수가 적은 후보자들 중에서 가장 먼저 추천된 후보자의 추천수를 0으로 만들고 액자틀에서 빼야한다. 그러므로, 각 후보자에게는 추천수와 등록된 순서를 입력해주어야 한다. 그래서 평소에는 액자틀 리스트에 학생 번호만 입력하고 2-2인 경우에만 탈락시킬 후보자를 찾기 위하여 새로운 리스트를 통하여 가장 추천수가 낮고 가장 먼저 입력된 후보자를 가려내는 작업을 추가하여 풀이하면 된다. 주의할 점은 추천 리스트를 확인하면서 이미 액자틀에 등록된 학생의 경우에는, 등록된 순서를 갱신하는 실수를 하여서는 안된다.

소스 코드

import sys
from collections import defaultdict
n = int(sys.stdin.readline())
st = int(sys.stdin.readline())
rec_st = defaultdict(int)
check_li = defaultdict(int)
li = list(map(int, sys.stdin.readline().strip().split()))
ba = []
for i in range(st):
    rec_st[li[i]] += 1
    if not li[i] in ba:
        check_li[li[i]] = i
        if len(ba) < n:
            ba.append(li[i])
        else:
            tp = []
            for j in ba:
                tp.append([rec_st[j], check_li[j], j])
            tp = sorted(tp, key=lambda x:(-x[0], -x[1]))
            rm_idx = tp[-1][2]
            rec_st[rm_idx] = 0
            ba.pop(ba.index(rm_idx))
            ba.append(li[i])
ba.sort()
print(*ba)