베스트앨범

프로그래머스 - 베스트앨범

문제 설명

스트리밍 사이트에서 장르 별로 가장 많이 재생된 노래를 두 개씩 모아 베스트 앨범을 출시하려 합니다. 노래는 고유 번호로 구분하며, 노래를 수록하는 기준은 다음과 같습니다.

1. 속한 노래가 많이 재생된 장르를 먼저 수록합니다.
2. 장르 내에서 많이 재생된 노래를 먼저 수록합니다.
3. 장르 내에서 재생 횟수가 같은 노래 중에서는 고유 번호가 낮은 노래를 먼저 수록합니다.

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

제한 사항

• genres[i]는 고유번호가 i인 노래의 장르입니다.
• plays[i]는 고유번호가 i인 노래가 재생된 횟수입니다.
• genres와 plays의 길이는 같으며, 이는 1 이상 10,000 이하입니다.
• 장르 종류는 100개 미만입니다.
• 장르에 속한 곡이 하나라면, 하나의 곡만 선택합니다.
• 모든 장르는 재생된 횟수가 다릅니다.

입출력 예

풀이 - 실패

def solution(genres, plays):
    music_group_by_genre = {}
    play_time_group_by_genre = {}
    answer = []

    for i in range(len(genres)):
        genre = genres[i]
        play_time = plays[i]

        play_time_group_by_genre[genre] = play_time_group_by_genre.get(genre, 0) + play_time
        music_group_by_genre[genre] = music_group_by_genre.get(genre, []) + [float(str(play_time) + f".{i}")]

    for genre in sorted(play_time_group_by_genre, key=lambda x: play_time_group_by_genre[x], reverse=True):
        music_list = sorted(music_group_by_genre[genre], reverse=True, key=lambda x: (int(x), -int(str(x)[-1])))
        end = 2 if len(music_list) > 1 else len(music_list)

        for i in range(end):
            answer.append(int(str(music_list[i])[-1]))

플레이 시간을 모두 합해 가장 높은 장르마다, 해당 장르에 속해 있는 곡의 재생 시간을 정렬하여 가장 높은 곡을 answer에 수록한다. 이때 고유 번호는 소수점 아래 첫 자리에 저장해놨기 때문에 이를 파싱해 저장하면 될 것이라 생각했다. 하지만 결과는 실패였고, 다른 해결 방법을 찾아야 했다.

다른 사람의 풀이

def solution(genres, plays):
    play_time_group_by_genre = {}
    music_group_by_genre = {}
    answer = []

    for i in range(len(genres)):
        genre = genres[i]
        play_time = plays[i]

        play_time_group_by_genre[genre] = play_time_group_by_genre.get(genre, 0) + play_time

        if genre in music_group_by_genre:
            music_group_by_genre[genre].append((i, play_time))
        else:
            music_group_by_genre[genre] = [(i, play_time)]

    genre_sorted_list = sorted(play_time_group_by_genre, key=lambda x: play_time_group_by_genre[x], reverse=True)

    for genre in genre_sorted_list:
        music_sorted_list = sorted(music_group_by_genre[genre], key=lambda x: (x[1], -x[0]), reverse=True)
        answer.append(music_sorted_list[0][0])

        if len(music_sorted_list) > 1:
            answer.append(music_sorted_list[1][0])

    return answer

결국, 다른 사람의 풀이를 참고하여 문제를 해결하였다. [프로그래머스] Lv. 3 - 베스트앨범 | 메시에님의 블로그에 있는 코드를 참고하여 해결하였다.

우선, 내 풀이의 접근법이 틀리지는 않았다. 다만 정렬 과정에서 뭔가 착오가 있었던 것 같다.

해결한 풀이도 결국 장르별로 총 플레이 시간을 구해 가장 많이 플레이 된 장르부터 순서대로 가져올 수 있도록 리스트를 만든 뒤, 해당 장르에 속해있는 곡들을 고유번호와 플레이 시간을 토대로 정렬한 뒤, answer에 추가해 반환한다.

이 코드에서 마음에 인상깊었던 부분은 배열의 첫 요소를 answer에 넣은 뒤, 배열 요소가 2 이상이라면 2번째 요소를 answer에 넣는 코드였다. 나는 처음에 길이가 2 이상이냐 아니냐로 동작을 나누려고 했는데, 이런 식으로 중복을 제거해 효율적인 코드를 짠 것이 굉장히 인상깊었다.

알게 된 것

다른 사람의 풀이를 보면서 defaultdict라는 것이 있다는 것을 알게 되었다.

딕셔너리를 만들어주며, 작동방식 또한 기존 딕셔너리와 거의 동일한데, 처음 키 값이 들어오면 해당 키에 할당해주는 초기값을 지정할 수가 있다. 그래서 내 코드에서 if genre in ... 이런 식으로 key가 있는지 없는지 검사하는 코드를 없앨 수 있다는 장점이 있다.