/ ALGORITHM

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

image-20230125195131667

from collections import defaultdict

def solution(genres, plays):
    music = defaultdict(lambda: [])
    for i in range(len(genres)):
        music[genres[i]].append([i,plays[i]])
    
    answer = []
    music = music.items()
    music = sorted(music,key=lambda x:sum(y[1] for y in x[1]),reverse=True)
    
    for i in music:
        i[1].sort(key=lambda x: (x[1],-x[0]),reverse=True)
        
        count=0
        for j in i[1]:
            if count==2:
                break
            answer.append(j[0])
            count+=1
            
    return answer