[Programmers] 해시 - 베스트앨범

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

프로그래머스

 

야호 ! 오랜만에 레벨 3을 혼자서 30분만에 풀었따.

작년 이맘때쯤 끙끙 답 참고하면서 풀었었는데, 그래도 실력이 늘긴 늘었나 보다.

 

 

문제 조건

- 속한 노래가 많이 재생된 장르를 먼저 수록합니다.

   -> 딕셔너리로 key : 장르, value : 고유번호 그리고 key : 장르, value : 장르에 속한 노래의 총 재생 횟수 를 만들어준다.
- 장르 내에서 많이 재생된 노래를 먼저 수록합니다.

  -> 딕셔너리로 key : 고유번호 value : 재생횟수 를 만들어준다. (첫 번째 조건으로 찾은 value값으로 key를 탐색해야 하기 때문)
- 장르 내에서 재생 횟수가 같은 노래 중에서는 고유 번호가 낮은 노래를 먼저 수록합니다.

  -> 이건 딕셔너리 이중 정렬로 해결한다.
노래의 장르를 나타내는 문자열 배열 genres와 노래별 재생 횟수를 나타내는 정수 배열 plays가 주어질 때, 베스트 앨범에 들어갈 노래의 고유 번호를 순서대로 return 하도록 solution 함수를 완성하세요.

 

삽질

1. 1번 조건을 총 재생 횟수가 아닌 속한 노래의 수로 잘못 이해

2. 장르에 속한 곡이 하나라면, 하나의 곡만 선택합니다. -> 이 조건 놓침

 

이 두개 때문에 시간을 좀 낭비했다.

문제를 풀 때 항상 모든 조건을 꼼꼼히 확인하자 !!

 

 

풀이

# 장르 별로 가장 많이 재생된 노래 두 개씩
# 1. 속한 노래가 많이 재생된 장르 먼저 (장르 선택)
# 2. 장르 내에서 많이 재생된 노래 (장르 내 노래 선택)
# 3. 재생수 같으면 고유 번호가 낮은 노래 먼저 (고유 번호로 선택)

def solution(genres, plays):
    answer = []
    genres_dict = {} # key : 장르 value : 고유번호(인덱스)
    plays_dict = {} # key : 고유번호(인덱스) value : 재생횟수
    genres_plays_dict = {} # key : 장르 value : 속한 노래들의 총 재생 수
    for idx in range(len(genres)):
        plays_dict[idx] = plays[idx]
        if genres_plays_dict.get(genres[idx]) == None:
            genres_plays_dict[genres[idx]] = plays[idx]
        else:
            genres_plays_dict[genres[idx]] += plays[idx]
        
        if genres_dict.get(genres[idx]) == None:
            genres_dict[genres[idx]] = [idx]
        else:
            genres_dict[genres[idx]].append(idx)
    
    genres_plays_dict = dict(sorted(genres_plays_dict.items(), key=lambda x:x[1], reverse=True))

    for genre in genres_plays_dict.keys(): # 한 장르당 두개 answer에 저장
        idx_plays = []
        for i in genres_dict[genre]:
            idx_plays.append([i, plays_dict[i]]) # [고유번호, 재생횟수] 저장
        idx_plays.sort(key=lambda x:(-x[1], x[0])) # 재생된 횟수 많은것, 재생된 횟수가 같다면 고유번호 작은것
        if len(idx_plays) == 1:
            answer.append(idx_plays[0][0])
        else:
            answer.append(idx_plays[0][0])
            answer.append(idx_plays[1][0])
    
    return answer

 

복습할 것

딕셔너리 정렬

genres_plays_dict = dict(sorted(genres_plays_dict.items(), key=lambda x:x[1], reverse=True))