[프로그래머스 lv1] 신고 결과 받기 - 파이썬(Python)

문제

프로그래머스

풀이

def solution(id_list, report, k):
    
    reports = {i : 0 for i in id_list} # 유저별 신고 받은 횟수
    res = {i : 0 for i in id_list} # 유저별 받을 메일 수
    
    # 유효한 신고들에 대하여, 유저별로, 신고받은 횟수 구하기 (즉, reports 구하기)
    for r in set(report):
        user, bad = r.split()
        reports[bad] += 1
        
    # 유효한 신고들에 대하여, 유저별로, 정지당한 유저를 신고한 횟수 구하기 (즉, res 구하기)
    for r in set(report):
        user, bad = r.split()
        if reports[bad] >= k:
            res[user] += 1
            
    return list(res.values())

결과