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

문제 링크 :https://programmers.co.kr/learn/courses/30/lessons/42579

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

---

My Solution

from collections import defaultdict
from collections import Counter


def solution(genres, plays):
    answer = []
    n = len(genres)
    times = defaultdict(dict)
    total = defaultdict(int)
    for idx, (genre, play) in enumerate(zip(genres, plays)):
        times[genre][idx] = play
        total[genre] += play

    for genre in Counter(total).most_common():
        for idx, time in Counter(times[genre[0]]).most_common(2):
            answer.append(idx)

    return answer

 

Key Idea

• Counter 를 이용해 최빈값 구하기
• total 로 장르 별 전체 값, times 로 장르별 최빈 idx를 관리.

Best solution 은 가독성, 간결함에서 이점이 없는 것 같아 첨부 X