명예의 전당 (1)

https://school.programmers.co.kr/learn/courses/30/lessons/138477

프로그래머스

[문제 설명]

"명예의 전당"이라는 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 함수를 완성해주세요.

 

제한사항

• 3 ≤ k ≤ 100
• 7 ≤ score의 길이 ≤ 1,000
  • 0 ≤ score[i] ≤ 2,000

[나의 풀이]

def solution(k, score):
    answer = []
    a = []
    for i in score:
        a.append(i)
        a = sorted(a, reverse = True)
        if len(a) < k:
            answer.append(a[-1])
        else:
            answer.append(a[k-1])
    return answer

 

[다른 풀이]

 

챗지피티에 따르면 내 풀이는 매번 리스트를 정렬하기 때문에, 입력 크기가 커질수록 성능이 저하된다고 한.

이 문제를 해결하기 위해서는 정렬을 최소화하거나 효율적인 자료구조를 사용하면 된다.

1)

import heapq

def solution(k, score):
    answer = []
    min_heap = []  # 최소 힙으로 사용할 리스트
    for s in score:
        heapq.heappush(min_heap, s)  # 값을 힙에 추가
        if len(min_heap) > k:  # 힙 크기가 k를 초과하면
            heapq.heappop(min_heap)  # 가장 작은 값을 제거
        answer.append(min_heap[0])  # 힙의 최솟값이 k번째 큰 값
    return answer

 

⭐heap(힙)이란 무엇인가?

https://littlefoxdiary.tistory.com/3

[Python] 힙 자료구조 / 힙큐(heapq) / 파이썬에서 heapq 모듈 사용하기

 

2)

from bisect import insort

def solution(k, score):
    answer = []
    a = []  # 명예의 전당 리스트 (항상 오름차순 정렬)

    for s in score:
        insort(a, s)  # 값을 정렬된 위치에 삽입
        if len(a) > k:  # 명예의 전당 크기가 k를 초과하면
            a.pop(0)  # 가장 작은 값 제거 (오름차순이므로 첫 번째 요소)
        answer.append(a[0])  # 현재 명예의 전당의 최하위 점수 추가

    return answer

insort는 Python의 bisect모듈에서 제공하는 함수로, 정렬된 리스트에 값을 효율적으로 삽입하는 데 사용됩니다.
이 함수는 리스트가 정렬된 상태를 유지하도록 값을 적절한 위치에 삽입하며, 자동으로 삽입 위치를 계산합니다.

 

3)

def solution(k, score):

    q = []

    answer = []
    for s in score:

        q.append(s)
        if (len(q) > k):
            q.remove(min(q))
        answer.append(min(q))

    return answer