조예지

def solution(genres, plays): g_list = {} #"genre" : [(재생횟수, 고유번호)] for i in range(len(genres)): if genres[i] not in g_list: g_list[genres[i]] = [] g_list[genres[i]].append((plays[i], i)) g_play = [] #(총재생횟수, 장르) for g in g_list: g_num = 0 for s_play in g_list[g]: g_num += s_play[0] g_play.append((g_num, g)) g_play.sort(reverse = True) answer = [] for i in range(len(g_play)): g = g_list[g_play[i][1]] g.sort(reverse = True) a = g[0] try: b = g[1] if a[0] == b[0]: answer.append(b[1]) answer.append(a[1]) else: answer.append(a[1]) answer.append(b[1]) except: answer.append(a[1]) return answer

... for i in range(len(g_play)): g = sorted(g_list[g_play[i][1]], key=lambda m: (-m[0], m[1])) answer.append(g[0][1]) try: answer.append(g[1][1]) except: pass return answer