https://programmers.co.kr/learn/courses/30/lessons/42579

베스트앨범.py

def solution(genres, plays):
    g_count = {}
    g_list = {}
    for idx in range(len(genres)):
        g = genres[idx]        
        g_count[g] = g_count.get(g, 0) + plays[idx]
        
        sg_list = g_list.get(g, [])
        sg_list.append((idx, plays[idx]))
        g_list[g] = sg_list
    
    g_order = sorted(g_count.items(), key=lambda x: x[1])
    g_order.reverse()
    g_order = [g[0] for g in g_order]

    answer = []
    for g in g_order:
        temp = sorted(g_list[g], key=lambda x: (x[1], -x[0]))
        temp.reverse()
        temp = [x[0] for x in temp][:2]
        answer += temp
    
    return answer