프로그래머스 베스트엘범 풀이 With Python

문제 링크: https://programmers.co.kr/learn/courses/30/lessons/42579

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

파이썬 자료구조인 Dictionary를 사용하여 장르별 재생 횟수와 장르내에서 각 노래별 재생 횟수를 Dictionary key, value 값으로 저장하여 가장 재생 횟수가 많은 장르 순서로, 장르 내에서는 재생 횟수가 높은 노래 순서로 2개 혹은 1개씩 answer 리스트에 담아서 답을 구합니다. 소스코드는 아래와 같습니다.

def solution(genres, plays):
    answer = []
    from collections import defaultdict
    l = len(genres)
    genrecount = defaultdict(int)
    music = defaultdict(list)
    for g in range(l):
        genrecount[genres[g]] += plays[g]
        music[genres[g]].append((g, plays[g]))
    gen = list(genrecount.items())
    gen.sort(key= lambda x: x[1], reverse=True)
    for g in gen:
        if len(music[g[0]]) == 1:
            answer.append(music[g[0]][0][0])
        else:
            music[g[0]].sort(key=lambda x: x[1], reverse=True)
            answer.append(music[g[0]][0][0])
            answer.append(music[g[0]][1][0])
    return answer

 소스코드에 대한 질문이 있으시면 댓글 남겨주시면 답변드리겠습니다. :-)