백준 1713. 후보 추천하기 (with Python)

문제

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

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

접근 방법

문제에서 제시한 제한사항들을 잘 준수해서 코드를 작성했다.

1. 크게 추천 학생 번호가 이미 사진 틀에 들어 있는 경우와 그렇지 않은 경우로 나누었다.
2. 이미 사진 틀에 있는 경우 그 번호의 학생의 투표수를 1개씩 올려주었다.
3. 반대로, 사진 틀에 없는 경우에는 다시 사진 틀에 사진이 주어진 개수를 넘는지 아닌지로 경우를 나누었다.
4. 넘는 경우에는 가장 앞에서부터 투표수를 순회하며 가장 작은 투표수에 해당하는 학생을 지웠다.
5. 넘지 않는 경우에는 사진틀에 append해주고 점수 리스트에도 append해주었다.
   (리스트는 사진 리스트와 점수 리스트로 나누어서 조작해주었다.)

코드

N = int(input())
vote = int(input())
students_num = list(map(int, input().split()))

photos = [] # 사진 리스트 
tmp = [] # 투표수 리스트

for i in range(vote):
	# 해당 학생의 사진이 이미 사진 틀에 있는 경우
    if students_num[i] in photos:
    	# 해당 학생을 찾아 투표수를 +1
        for j in range(len(photos)):
            if students_num[i] == photos[j]:
                tmp[j] += 1
    # 사진틀에 없는 경우
    else:
    	# 사진 틀 개수가 꽉 찼다면
        if len(photos) >= N:
        	# 최소값을 찾아서 먼저 등록된 학생 삭제
            for j in range(N):
                if tmp[j] == min(tmp):
                    del photos[j]
                    del tmp[j]
                    break
        # 아니라면 뒤에 그냥 추가
        photos.append(students_num[i])
        tmp.append(1)

photos.sort()
print(' '.join(map(str,photos)))