반응형
def solution(genres, plays):
answer = []
playDic = {}
dic = {}
for i in range(len(genres)):
playDic[genres[i]] = playDic.get(genres[i], 0) + plays[i]
dic[genres[i]] = dic.get(genres[i], []) + [(plays[i], i)]
genreSort = sorted(playDic.items(), key = lambda x: x[1], reverse = True)
for (genre, totalPlay) in genreSort:
dic[genre] = sorted(dic[genre], key = lambda x: (-x[0], x[1])) #내림차순, 오름차순
answer += [idx for (play, idx) in dic[genre][:2]]
return answer
반응형
'알고리즘 > 프로그래머스 level 3' 카테고리의 다른 글
[파이썬🐍] 프로그래머스 : 정수 삼각형 (0) | 2021.07.19 |
---|---|
[파이썬🐍] 프로그래머스 : 등굣길 (0) | 2021.07.18 |
[파이썬🐍] 프로그래머스 : 네트워크 (0) | 2021.07.17 |
[파이썬🐍] 프로그래머스 : 가장 먼 노드 (0) | 2021.06.30 |
[파이썬🐍] 프로그래머스 : 입국심사 (0) | 2021.06.25 |
댓글