[프로그래머스 Lv.3] 베스트앨범 (Python)

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

프로그래머스

풀이

장르별 플레이 누적 합을 담고 있는 딕셔너리와 장르별 음악 각각의 플레이 수를 담고 있는 딕셔너리 2개를 활용하여 문제를 풀이하면 쉽게 풀이할 수 있다. 레벨 3 문제치고 어렵지 않은 문제라고 생각한다.

from collections import defaultdict
from functools import cmp_to_key

def cmp(x, y): # 비교 함수
    if x[1] == y[1]: # 노래의 플레이 수가 같으면
        return x[0] - y[0] # 작은 번호가 먼저 나오도록
    return y[1] - x[1] # 플레이 수가 큰 것이 먼저 나오도록
    
def solution(genres, plays):
    play_cnt = defaultdict(int) # 장르별 플레이 누적 합을 담고 있는 딕셔너리
    musics = defaultdict(list) # 장르별 음악 각각의 플레이 수를 담고 있는 딕셔너리
    
    for i in range(len(genres)):
        genre = genres[i] # i번째 음악의 장르
        play = plays[i] # i번째 음악의 플레이 수
        
        play_cnt[genre] += play # 장르별 플레이 수를 누적
        musics[genre].append((i, play)) # 장르별 음악 각각의 플레이 수 저장
    
    for music_list in musics.values(): # 장르별 음악 각각을 순회하며
        music_list.sort(key=cmp_to_key(cmp)) # 플레이 수의 내림차순으로, 플레이 수가 같다면 번호의 오름차순으로 정렬
    
    answer = []
    for item in sorted(play_cnt.items(), key=lambda x: x[1], reverse=True): # 누적 플레이 수의 내림차순으로 장르를 정렬
        genre, _ = item
        song_ranks = musics[genre] # 장르별 음악 각각의 순위
        if len(musics[genre]) == 1: # 장르별 음악이 단 하나만 존재한다면
            answer.append(song_ranks[0][0]) # 그 음악의 번호를 저장
        else: # 하나 이상이라면
            answer.append(song_ranks[0][0]) # 가장 플레이 수가 높은 것 저장
            answer.append(song_ranks[1][0]) # 그 다음으로 플레이 수가 높은 것 저장
    return answer