level 1 신고 결과 받기

import를 해야하는지 몰라서 다른사람들의 풀이를 보고 다시 풀었다.

 

from collections import defaultdict

def soultion(id_list, report, k):
   
    answer = []
    
    to_from = defaultdict(set) 
    from_to = defaultdict(set) 


    for r in report:
        r_from, r_to = r.split(' ')
     
        to_from[r_to].add(r_from)
        from_to[r_from].add(r_to)

    
    for i in id_list:
        cnt = 0
        for j in from_to[i]:
            if len(to_from[j])>=k:
                cnt+=1
        answer.append(cnt)
    return answer