[프로그래머스] 베스트앨범 python (*)

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

출처 : 프로그래머스

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

---

def solution(genres, plays):
    answer = []
    arr = sorted([[genres[i], plays[i], i] for i in range(len(genres))],key = lambda x:(x[0],-x[1], x[2]))

    # 재생횟수
    total_play = {}
    for i in arr:
        if i[0] not in total_play:
            total_play[i[0]] = i[1]
        else:
            total_play[i[0]] += i[1]
    total_play = sorted(total_play.items(), key = lambda x:-x[1]) # 재생횟수가 더 많은 타입부터

    for t in total_play: 
        cnt = 0
        for i in arr:
            if t[0] == i[0]: # 재생횟수 많은 노래 타입과 arr의 타입 같다면 
                cnt += 1
                if cnt > 2: # 한 장르에 앨범 2개만 실어야해서
                    break
                else:
                    answer.append(i[2]) # 노래의 고유번호 삽입
    return answer

풀이1

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

    for (k,v) in sorted(dict2.items(), key = lambda x:-x[1]): # 많이 재생한 노래 타입 먼저
        for (i,p) in sorted(dict1[k], key = lambda x:-x[1])[:2]: # 많이 재생한 노래 2개
            answer.append(i)

    return answer

풀이2

 

1. 속한 노래 많이 재생된 장르

   2. 장르 내 가자아 많이 재생된 노래

        3. 같으면 고유 번호 낮은 노래 먼저

 

우선순위를 생각하자.