susinlee 님의 블로그

[문제]

코니는 영어 단어가 적힌 카드 뭉치 두 개를 선물로 받았습니다. 코니는 다음과 같은 규칙으로 카드에 적힌 단어들을 사용해 원하는 순서의 단어 배열을 만들 수 있는지 알고 싶습니다.

• 원하는 카드 뭉치에서 카드를 순서대로 한 장씩 사용합니다.
• 한 번 사용한 카드는 다시 사용할 수 없습니다.
• 카드를 사용하지 않고 다음 카드로 넘어갈 수 없습니다.
• 기존에 주어진 카드 뭉치의 단어 순서는 바꿀 수 없습니다.

예를 들어 첫 번째 카드 뭉치에 순서대로 ["i", "drink", "water"], 두 번째 카드 뭉치에 순서대로 ["want", "to"]가 적혀있을 때 ["i", "want", "to", "drink", "water"] 순서의 단어 배열을 만들려고 한다면 첫 번째 카드 뭉치에서 "i"를 사용한 후 두 번째 카드 뭉치에서 "want"와 "to"를 사용하고 첫 번째 카드뭉치에 "drink"와 "water"를 차례대로 사용하면 원하는 순서의 단어 배열을 만들 수 있습니다.

문자열로 이루어진 배열 cards1, cards2와 원하는 단어 배열 goal이 매개변수로 주어질 때, cards1과 cards2에 적힌 단어들로 goal를 만들 있다면 "Yes"를, 만들 수 없다면 "No"를 return하는 solution 함수를 완성해주세요.

[풀이]

goal의 요소를 하나씩 가져와서 cards1 과 cards2 의 첫번째 요소와 비교한 후, 같으면 해당 요소를 삭제하고 같지 않으면 'No'를 리턴하면 될듯하다. goal의 모든 요소를 전부 클리어하면 'Yes'를 리턴하자. 첫번째 요소를 제거해야 하니 deque를 쓰는 것이 좋아보인다.

from collections import deque

def solution(cards1, cards2, goal):
    
    cards1 = deque(cards1)
    cards2 = deque(cards2)
    
    for word in goal:
        
        if cards1 and word == cards1[0]:
            cards1.popleft()
        elif cards2 and word == cards2[0]:
            cards2.popleft()
        else:
            return 'No'
    
    return 'Yes'

[문제]

"명예의 전당"이라는 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