[Python] 해시_베스트앨범

💡 문제

해시_베스트앨범 문제 링크

✅ 풀이

def solution(genres, plays):
    
    dic = {}
    playDic = {}
    answer = []
    
    for i in range(len(genres)):
        playDic[genres[i]] = playDic.get(genres[i], 0) + plays[i]
        dic[genres[i]] = dic.get(genres[i], []) + [(plays[i], i)]
        
    # 속한 노래가 많이 재생된 장르 순서
    sort_genres = sorted(playDic.items(), key = lambda x : x[1], reverse = True)
    
    
    for (genre, play) in sort_genres:
        dic[genre] = sorted(dic[genre], key = lambda x : (-x[0], x[1]))
        answer += [i[1] for i in dic[genre][:2]]

    
    return answer

🟩 알아두기