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

def solution(genres, plays):
    playlist = {}
    count = {}
    for idx, (g, p) in enumerate(zip(genres, plays)):
        if g in playlist:
            playlist[g].append((p,idx))
            count[g] += p
        else:
            playlist[g] = [(p,idx)]
            count[g] = p
            
    ans = []
    for g in sorted(count.items(), reverse=True, key=lambda x:x[1]):
        for p in sorted(playlist[g[0]], reverse=True, key=lambda x:x[0])[:2]:
            ans.append(p[1])
    
    return ans

def solution(genres, plays):
    answer = []
    d = {e:[] for e in set(genres)}
    for e in zip(genres, plays, range(len(plays))):
        d[e[0]].append([e[1] , e[2]])
    genreSort =sorted(list(d.keys()), key= lambda x: sum( map(lambda y: y[0],d[x])), reverse = True)
    for g in genreSort:
        temp = [e[1] for e in sorted(d[g],key= lambda x: (x[0], -x[1]), reverse = True)]
        answer += temp[:min(len(temp),2)]
    return answer