명예의 전당1

[문제]

"명예의 전당"이라는 TV 프로그램에서는 매일 1명의 가수가 노래를 부르고, 시청자들의 문자 투표수로 가수에게 점수를 부여합니다. 매일 출연한 가수의 점수가 지금까지 출연 가수들의 점수 중 상위 k번째 이내이면 해당 가수의 점수를 명예의 전당이라는 목록에 올려 기념합니다. 즉 프로그램 시작 이후 초기에 k일까지는 모든 출연 가수의 점수가 명예의 전당에 오르게 됩니다. k일 다음부터는 출연 가수의 점수가 기존의 명예의 전당 목록의 k번째 순위의 가수 점수보다 더 높으면, 출연 가수의 점수가 명예의 전당에 오르게 되고 기존의 k번째 순위의 점수는 명예의 전당에서 내려오게 됩니다.

 

이 프로그램에서는 매일 "명예의 전당"의 최하위 점수를 발표합니다. 예를 들어, k = 3이고, 7일 동안 진행된 가수의 점수가 [10, 100, 20, 150, 1, 100, 200]이라면, 명예의 전당에서 발표된 점수는  [10, 10, 10, 20, 20, 100, 100]입니다.

 

명예의 전당 목록의 점수의 개수 k, 1일부터 마지막 날까지 출연한 가수들의 점수인 score가 주어졌을 때, 매일 발표된 명예의 전당의 최하위 점수를 return하는 solution 함수를 완성해주세요.

 

[풀이]

score 를 하나씩 가져와서 자료구조에 넣고

자료구조의 길이가 k를 넘으면 작은 값을 빼내고

그렇지 않으면 작은 값을 발표하는 방법

으로 코드를 구현하면 될 듯 하다,

 

파이썬에서 제공하는 heapq (최소 힙)이라는 자료구조를 알면 효율적으로 풀 수 있다.

주요 메서드는 아래와 같다

1. heappush(heap, e) - 요소를 힙에 추가 O(log n)

2. heappop(heap) - 힙에서 가장 작은 요소를 제거하고 반환 O(log n)

3. heappushpop(heap, e) - 요소를 힙에 추가한 후, 가장 작은 요소를 제거하고 반환 O(log n)

4. heapreplace(heap, e) - 힙의 가장 작은 요소를 제거, 새로운 요소를 추가 O(log n) → 제거 후 추가라 힙의 크기 유지

5. heapify(iterable) - 리스트를 힙으로 변환 O(n)

6. nlargest(n, iterable) & nsmallest(n, iterable) - 가장 큰 요소 또는 작은 요소를 찾아 반환 O(n log k)

 

import heapq

def solution(k, scores):
    
    honor = []
    prese = []
    
    for score in scores:
        heapq.heappush(honor, score)
        if len(honor) > k:
            heapq.heappop(honor)
        prese.append(honor[0])
    
    return prese