[해시] 베스트앨범

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

from collections import defaultdict
 
def solution(genres, plays):
    answer = []
    d = defaultdict(list)
    p = defaultdict(lambda: [0, 0])
 
    for i in range(len(genres)):
        d[genres[i]].append((i, plays[i]))
    # print(d)  # {'classic': [(0, 500), (2, 150), (3, 800)], 'pop': [(1, 600), (4, 2500)]})
 
    # [장르의 수, 장르의 총 재생횟수]
    for i, j in zip(genres, plays):
        p[i][0] += 1
        p[i][1] += j
    a = sorted(p.items(), key=lambda x: -x[1][1])
    prior = []
    for i in a:
        prior.append(i[0])
    # print(prior)  # ['pop', 'classic']
 
    for i in prior:
        temp = list(d[i])
        s = sorted(temp, key=lambda x: -x[1])
 
        if len(s) >= 2:
            answer.append(s[0][0])
            answer.append(s[1][0])
        else:
            answer.append(s[0][0])
    return answer
 
 
g = ["classic", "pop", "classic", "classic", "pop"]
p = [500, 600, 150, 800, 2500]
print(solution(g, p))