[프로그래머스/Lv.1/Python] 베스트앨범

def solution(genres, plays):
    answer = []
    total = {} # 장르 : 총재생횟수
    genDic = {} # 장르 : (재생횟수, 고유넘버)

    for i in range(len(genres)):
        genre = genres[i]
        play = plays[i]
        if genres[i] in total.keys():
            total[genres[i]] += plays[i]
            genDic[genres[i]].append((plays[i],i))
        else:
            total[genres[i]] = plays[i]
            genDic[genre] = [(play,i)]

    # 가장 재생횟수가 높은 장르
    total = sorted(total.items(), key=lambda x: x[1], reverse=True)

    for key in total:
        playlist = genDic[key[0]]
        playlist = sorted(playlist, key=lambda x: x[0], reverse=True)
        print(playlist)
        for i in range(len(playlist)):
            if i==2:
                break
            answer.append(playlist[i][1])

    print(playlist)

    return answer