[파이썬] [해시] Pro 베스트앨범

문제

https://programmers.co.kr/learn/courses/30/lessons/42579?language=python3 

코딩테스트 연습 - 베스트앨범

 

코드

def solution(genres, plays):
    dict={}
    result=[]
    for i in range(len(genres)):
        if not genres[i] in dict:
            dict[genres[i]]=plays[i]
        else:
            dict[genres[i]]+=plays[i]
    #{'classic': 1450, 'pop': 3100}
    dict1=sorted(dict.items(), key=lambda x: x[1], reverse=True)
    #[('pop', 3100), ('classic', 1450)]
    for i in dict1:
        temp={}
        for j in range(len(genres)):
            if genres[j]==i[0]:
              temp[j]=plays[j]
        temp1=sorted(temp.items(), key=lambda x: x[1], reverse=True)
        result.append(temp1[0][0])
        if len(temp1)>1:
            result.append(temp1[1][0])
    #[4, 1, 3, 0]
    return result