[programmers/ 고득점 kit] 해시: 베스트 앨범

dictionary data type을 활용하여 푸는 문제 

 

python에서 stl의 compare와 같은 구현을 통하여 원하는 방식으로 sort할 수 있다.

python 3.x 버전 부터는 sorted 인자로 key = cmp_to_key(보조 함수이름)를 통하여 구현 가능.

dict를 sort 할 때 sorted(dict, ...) 로 구현하면 key만 가지는 list가 리턴되니 조심하자.

 

아래에는 1.문제링크, 2.code, 3.테스트 케이스 순으로 첨부하였다.

 

1.문제링크

 

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

알고리즘 연습 - 베스트앨범 | 프로그래머스

 

2.code :

 

from functools import cmp_to_key

def solution(genres, plays):
    answer = []
    
    d_score = {}
    d_index = {}
    for index, (genres,plays) in enumerate(zip(genres, plays)):
        d_score[genres] = d_score.get(genres, 0) + plays
        
        if d_index.get(genres, 0) == 0 :
            d_index[genres] = []
        
        d_index[genres].append([index, plays])
    
    print(d_score)
    # print(d_index)
    
    l_sorted = sorted(d_score.items(), key = lambda k : k[1], reverse =True)
    print("l_sorted: ",l_sorted)
    
    for obj in l_sorted:
        genres = obj[0]
        l_srtd_plays = sorted(d_index[genres],key= cmp_to_key(compare))
        
        answer.append(l_srtd_plays[0][0])
        if len(l_srtd_plays) >1:
            answer.append(l_srtd_plays[1][0])
        
        print(genres, l_srtd_plays)
    
        
    return answer

def compare(x,y):
    if x[1] == y[1]:
        if x[0] < y[0] :
            return -1
        else:
            return 1
    
    elif x[1] < y[1]:
        return 1
    else:
        return -1

 

 

3.테스트 케이스