[python] 프로그래머스 92334번 : 신고 결과 받기

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

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

 

구현 

from collections import defaultdict

def solution(id_list, report, k):
    # 중복 제거 
    report = list(set(report))
    result = [0 for _ in range(len(id_list))]
    # 고소 당한 사람의 고소 당한 횟수 카운트 
    accused = defaultdict(int)
    for data in report:
        p = data.split()[1]
        accused[p] += 1
    # 정지당한 사람들 리스트 
    paused = [key for key, v in accused.items() if v >= k]
    # 정지당한 사람들을 고소한 사람들 횟수 증가
    for data in report:
        p1, p2 = data.split()
        if p2 in paused:
            idx = id_list.index(p1)
            result[idx] += 1
    return result