[프로그래머스] 베스트앨범 (python)

링크

https://programmers.co.kr/learn/courses/30/lessons/42579?language=python3

코딩테스트 연습 - 베스트앨범

풀이

 

일단 python으로 언어를 바꾼지 얼마 안되.. 코드의 상태가 좋지 못하는점 양해 부탁드립니다..

 

노래의 장르를 나타내는 문자열 배열 genres와 노래별 재생 횟수를 나타내는 정수 배열 plays가 주어질 때, 베스트 앨범에 들어갈 노래의 고유 번호를 순서대로 구하는 문제 이다.

 

1. 먼저 dictionay를 이용해서 같은 장르의 곡의 합을 먼저 구하고

 

2. 각 장르에 속한 곡들의 재생 횟수를 내림차순으로 정렬(pq 이용)을 하고 최대 2개를 리턴 해야한다.

 

 

from queue import PriorityQueue

def solution(genres,plays):
    janru = {}
    song = {}

    for i in range(len(genres)):
        if not bool(janru.get(genres[i])) :
            janru[genres[i]] = plays[i]
            song[genres[i]] = PriorityQueue()
            song[genres[i]].put((-plays[i],i))
        else:
            janru[genres[i]] += plays[i]
            song[genres[i]].put((-plays[i],i))

    janru = dict(sorted(janru.items(), key =(lambda janru: janru[1]),reverse=True))

    answer = []

    for name in janru:
        print(name)
        for cnt in range(2):
            if song[name].empty():
                break
            answer.append((song[name].get())[1])
    return answer