(프로그래머스) 코딩테스트 연습 - 해시- 베스트앨범 (Python)

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

코딩테스트 연습 - 베스트앨범 | 프로그래머스

문제 설명

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

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

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

제한사항

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

 

제가 통과한 코드

from collections import defaultdict

def solution(genres, plays):
    answer = []
    
    dict_genres = defaultdict(list) #장르별 재생횟수를 담는 dictionary
    genres_hap = {} #장르별 합을 담는 dictionary
    
    max_rank = [] #장르의 합중 큰 순대로 담는 list
    
    dict_id = defaultdict(list) #장르별 재생횟수에 따른 id를 담는 dictionary
    
    # 장르별 재생횟수
    for genre, play in zip(genres, plays):
        dict_genres[genre].append(play)
    
    # 장르별 합
    for genre in dict_genres.keys():
        genres_hap[genre] = sum(dict_genres[genre])
    
    # 장르별 rank list
    max_rank = sorted(genres_hap, key = lambda x: genres_hap[x], reverse = True)
    
    # 장르별 id
    for i, genre in enumerate(genres):
        dict_id[genre].append(i)
    
    # main method
    # list에서 max 순대로 key값을 빼서 하나씩 추가할 수 있도록 함
    for k in max_rank:
        temp = {}
        
        # id를 key로 하여 각 재생횟수의 id를 결합하여 줌
        for v, k in zip(dict_genres[k], dict_id[k]):
            temp[k] = v
        
        # id와 재생횟수를 매칭 시킨 후 재생횟수가 큰 순서대로 정렬
        temp = sorted(temp, key=lambda x:temp[x], reverse=True)
        
        # 큰 순서대로 정렬하였으므로 앞의 2개만 answer list에 추가
        for i in temp[0:2]:
            answer.append(int(i))
    
    return answer

풀이

4가지의 그릇을 만든 후 조합하여 하나의 결과를 뽑아낸다라는 느낌으로 알고리즘을 작성하였습니다.

그 외의 내용은 주석을 보면 이해가 가능하리라 생각합니다.

 

느낀점

def등 으로 조금더 간결하게 정리할 수 있지만, 이런 원시적으로 접근하는 알고리즘을 보여주고 싶었습니다. 

이번 프로그래머스의 연습문제를 다 풀고 그 다음부터는 def와 class를 활용하여 간결하게 작성하는 방법을 연습해야겠습니다. 하지만 아직까지 갈 길이 멀지요,,

 

이 문제는 워낙 사람들간에도 다양하게 문제를 푼 것 같습니다. 후에 시간이 된다면 다른사람의 코드들을 들여다보는 것도 도움이 될 것 같습니다.