프로그래머스 베스트앨범 파이썬

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

코딩테스트 연습 - 베스트앨범

 

def solution(genres, plays):
    answer = []
    dic = {}
    for i in range(len(genres)):
        genre = genres[i]
        if genre not in dic:
            dic[genre] = [plays[i], (plays[i], i)]
        else:
            dic[genre][0] += plays[i]
            dic[genre].append((plays[i], i))

    temp = []
    for d in dic.values():
        temp.append(d)

    temp.sort(reverse=True)

    temp1 = []
    for t in temp:
        temp1.append(sorted(t[1:], key=lambda x: (x[0], -x[1]), reverse=True))

    for t in temp1:
        if len(t) == 1:
            answer.append(t[0][1])
        else:
            answer.append(t[0][1])
            answer.append(t[1][1])

    return answer


print(solution(["classic", "pop", "classic", "classic", "pop"], [500, 2500, 150, 800, 2500]))

# ["classic", "pop", "classic", "classic", "pop"]
# [500, 600, 150, 800, 2500]
# [4, 1, 3, 0]

 

이 문제는 해시를 이용한 구현유형에 속하는 문제인 듯 하다.

 

난이도 자체는 개인적으로 매우 쉬움 정도 로 느껴졌다.

 

조건:

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

풀이:

 

1. 장르 배열을 순회한다.

 

2. 장르 배열의 장르명을 키로 플레이 횟수를 값으로 딕셔너리에 저장한다.

 

3. 장르 총합을 딕셔너리 값의 첫번째 요소로 저장한다.

 

4. 이제부터 딕셔너리의 키는 의미가 없다고 판단하여 없애주었고, 총합이 많은 장르의 곡을 우선 수록해야하므로 첫번째 요소(장르별 재생 총합) 를 기준으로 내림차순 정렬한다.

 

5. 장르별 노래의 재생횟수를 내림차순으로 정렬하는데 이때 재생 횟수가 같은 경우 고유번호가 낮은 번호를 수록하기 위해 고유번호는 음수 정렬 처리 즉, 오름차순 정렬 한다.

 

6. 최종 정렬된 배열의 요소를 결과값에 추가한다.

 

 

PS. 코드가 좀 너저분한데 파이썬의 편리한 문법을 잘 모르기때문이다.

다른 분들의 코드를 참고하니 파이썬의 문법을 사용하니 코드가 무척 세련되었다.

 

아래는 개선한 코드

def solution(genres, plays):
    answer = []
    dic = {}
    dic2 = {}
    for i, (g, p) in enumerate(zip(genres, plays)):
        if g in dic:
            dic[g].append((i, p))
        else:
            dic[g] = [(i, p)]

        if g in dic2:
            dic2[g] += p
        else:
            dic2[g] = p

    for (k, _) in sorted(dic2.items(), key=lambda x: x[1], reverse=True):
        for (i, p) in sorted(dic[k], key=lambda x: (x[1], -x[0]), reverse=True)[:2]:
            answer.append(i)

    return answer


print(solution(["classic", "pop", "classic", "classic", "pop"], [500, 2500, 150, 800, 2500]))

# ["classic", "pop", "classic", "classic", "pop"]
# [500, 600, 150, 800, 2500]
# [4, 1, 3, 0]