[프로그래머스] 베스트앨범 - 파이썬(Python)

[프로그래머스] 베스트앨범 - 파이썬(Python)

 

 

 

from collections import defaultdict

def solution(genres, plays):
    answer = []
    G = defaultdict(int)
    detail = defaultdict(list)
    for idx, [g, p] in enumerate(zip(genres, plays)):
        G[g] += p
        detail[g].append((p, idx))
    for g, _ in sorted(G.items(), key=lambda x: -x[1]):
        for _, i in sorted(detail[g], key=lambda x: -x[0])[:2]:
            answer.append(i)
    return answer

혹은

from collections import defaultdict

def solution(genres, plays):
    answer = []
    G = defaultdict(list)
    for idx, [g, p] in enumerate(zip(genres, plays)):
        G[g].append((idx, p))
    order = sorted(G.items(), key=lambda x: sum(dict(x[1]).values()), reverse=True)
    for _, L in order:
        for i, _ in sorted(L, key=lambda x: -x[1])[:2]:
            answer.append(i)
    return answer

 

 

 

 

 

https://programmers.co.kr/learn/courses/30/lessons/42579

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