[프로그래머스] Hash level3 베스트앨범

프로그래머스의 코딩연습 중 level3의 hash 카테고리의 "베스트앨범" 문제 풀이

(문제 : 아래 링크👇👇👇)

프로그래머스

 

문제풀이

문제가 엄청 어렵지는 않은데.. 내 생각에는 설명이 친절하지 않았다고 생각한다.

일단 내가 헷갈렸던 지점은

1. 전체 곡들 중에서 조회수가 가장 많은 곡이 해당하는 장르부터 앨범에 담는다
2. 장르별로 조회수를 합하여 가장 조회수가 높은 장르부터 앨범에 담는다

반응형

(아무리 문제 설명을 읽어봐도 너무 모호하게 써있는 것 같았음)

처음에 1번으로 생각하고 문제를 풀었더니 1,2,3,4,15를 제외하곤 모두 실패가 떴다. 런타임에러도 아니고..

입출력 예시에서 보면 굳이 장르별 조회수 합을 보여주는 걸 보고 2번으로 접근해봤다. 

그랬더니 테스트케이스 모두 통과함......

뭔가 굉장히 허무하다 허허

 

코드는 다음과 같다.

######### 정확성 100% #########
def solution(genres, plays):
    # initialize
    total_cnt = {genre: 0 for genre in set(genres)} 
    sorted_dic = {genre: [] for genre in set(genres)}
    idx = range(len(genres))
    ziped = list(zip(plays,genres,idx))
    # fill in dict
    for cnt,key,idx in ziped :
        total_cnt[key] += cnt
    # sort total count dict by value
    total_cnt = {k: v for k, v in sorted(total_cnt.items(), key=lambda item: -item[1])}
    for cnt,key,idx in ziped :
        sorted_dic[key].append((cnt,idx))
    res = []
    for key in total_cnt :
        res.extend([x[1] for x in sorted(sorted_dic[key],key = lambda x:(-x[0],x[1]))[:2]])
    return res

• total_cnt : 장르별 조회수 합을 구하기 위한  dictionary. (key: 장르, value: 장르의 조회수 합)
• sorted_dic : 장르별 각 곡의 정보(조회수,고유번호)를 가지고 있는 dictionary. (key: 장르, value: 장르에 해당하는 곡들의 (조회수,고유번호) 리스트)

 

내가 제시한 코드가 도움이 됐기를 바라면서..!! 마친다 😊