[프로그래머스 코테 연습] 베스트앨범

문제 요약

노래는 고유 번호로 구분하며, 노래를 수록하는 기준은 다음과 같습니다.
1. 속한 노래가 많이 재생된 장르를 먼저 수록합니다.
2. 장르 내에서 많이 재생된 노래를 먼저 수록합니다.
3. 장르 내에서 재생 횟수가 같은 노래 중에서는 고유 번호가 낮은 노래를 먼저 수록합니다.
노래의 장르를 나타내는 문자열 배열 genres와 노래별 재생 횟수를 나타내는 정수 배열 plays가 주어질 때, 베스트 앨범에 들어갈 노래의 고유 번호를 순서대로 return 하도록 solution 함수를 완성하세요.

답안 예시

from collections import defaultdict

def solution(genres, plays):
    answer = []
    genre_dic = defaultdict(lambda: 0)
    total_list = defaultdict(lambda: [])
    
    for i in range(len(genres)):
    # 장르별 재생 수를 담은 dict 생성
        genre_dic[genres[i]] += plays[i]
  
    # 장르를 key로 갖고 value로 (곡 재생 수, 곡 넘버)를 갖는 dict 생성
        total_list[genres[i]].append((plays[i], i))
        
    # dic.items() 는 key와 value를 같이 출력
    # 장르별 재생 수를 기준으로 오름차순 정렬
    genre_dic = sorted(genre_dic.items(), key=lambda x:x[1], reverse=True)
    
    # total_list 각 key(장르)를 곡 재생 수 기준으로 오름차순 정렬
    # 장르별로 가장 많이 재생된 노래를 2개씩 모은다고 했으므로, [:2]로 자름
    for i in total_list:
        total_list[i] = sorted(total_list[i], key=lambda x:x[0], reverse=True)[:2]
    
    # 장르가 1개 남을 동안(모든 장르를 분석할 동안)
    while len(genre_dic) > 0:
        # 가장 많이 재생된 장르를 꺼냄
        genre_max = genre_dic.pop(0)
        # total_list 의 key(장르) 별로
        for t in total_list:
            if t == genre_max[0]:
                # 장르에 담긴 곡이 1개 초과면
                if len(total_list[t]) > 1:
                    # answer에 장르[가장 재생 수 많은 곡][곡 넘버] 담음
                    answer.append(total_list[t][0][1])
                    # answer에 장르[두 번째로 재생 수 많은 곡][곡 넘버] 담음
                    answer.append(total_list[t][1][1])
                else:
                    # 장르에 담긴 곡이 하나라면 하나의 곡만 선택
                    answer.append(total_list[t][0][1])
    return answer

느낀점

• defaultdict 에 대해 배웠음.
  
  
• 딕셔너리.items() 는 key 와 value 쌍을 함께 반환한다.
• key 값으로는 불변하는 값은 모두 가능하므로, 튜플도 올 수 있다.
• vlaue 값으로는 리스트도 올 수 있다.
• zip() 안의 range() 예시

   list(range(3))
   # [0, 1, 2]
   list(zip(range(3), ‘abc’))
   # [(0,‘a’), (1,‘b’), (2,‘c’)]