[프로그래머스] 신고 결과 받기 (Python)

 

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

코딩테스트 연습 - 신고 결과 받기

딕셔너리를 활용하는 문제다 뭔가 더 깔끔하게 풀 수 있을 것 같긴 하다.

 

나의 풀이

from collections import defaultdict
def solution(id_list, report, k):
    # 신고 당한 횟수 기록
    # 해당 유저 신고한 사람 기록
    # 신고 한 리스트
    report_dict = defaultdict(list)
    report_count = defaultdict(int)
    answer = []
    for rep in set(report):
        user, dic = rep.split()
        report_dict[user].append(dic)
        report_count[dic] +=1
    for id in id_list:
        recount = 0
        for user in report_dict[id]:
            if report_count[user] >= k:
                recount += 1
        answer.append(recount)
    return answer