[프로그래머스/파이썬]베스트 앨범

def solution(genres, plays):
answer = []
a={}
b={}
for i,(g,p) in enumerate(zip(genres,plays)):
if g not in a:
a[g]=[(i,p)]
else:
a[g].append((i,p))

    if g not in b:
        b[g]=p
    else:
        b[g]+=p
for (k,v) in sorted(b.items(),key=lambda x:x[1],reverse=True):
    for (i,p) in sorted(a[k],key=lambda x:x[1], reverse=True)[:2]:
        answer.append(i)
return answer

genres=["classic", "pop", "classic", "classic", "pop"]
plays=[500, 600, 150, 800, 2500]
print(solution(genres,plays))