[프로그래머스] 30. 베스트 앨범 - Python

프로그래머스 - 베스트 앨범

 

문제집 링크 : http://aladin.kr/p/szPra

내일은 코딩테스트 with 파이썬

프로그래머스 URL :  

[도서실습] 내일은 코딩테스트 with 파이썬(자료구조와 알고리즘의 기초부터 실전까지)

---

■ 작성 코드

from collections import defaultdict

def solution(genres, plays):
    
    check = defaultdict(list)
    play_check = defaultdict(int)
    
    for idx, (g, play) in enumerate(zip(genres, plays)):
        check[g].append([idx, play])
        play_check[g] += play
    
    sorted_plays = sorted(play_check.items(), key=lambda x:-x[1])
    
    answer = []
    
    for g, _ in sorted_plays:
        songs = sorted(check[g], key=lambda x: -x[1])[:2]
        answer.extend([song for song, _ in songs])
        
    
    
    return answer