-
마크다운 확인
확인 출처:프로그래머스 https://school.programmers.co.kr/learn/courses/30/lessons/42579 내 풀이
from collections import defaultdict def solution(genres, plays): g = defaultdict(int) g_num = defaultdict(list) for i in range(len(genres)): g[genres[i]] += plays[i] g_num[genres[i]].append((plays[i], i)) check = [] for k in g_num.keys(): g_num[k].sort(key=lambda x: (-x[0], x[1])) check.append((g[k], k)) check.sort(reverse=True) result = [] for i, j in check: n = 0 for count, num in g_num[j]: result.append(num) n += 1 if n == 2: break
<<
1
>>