[python] 프로그래머스_신고 결과 받기 (구현)

문제

 

풀이

 

def solution(id_list, report, k):
  answer = [0]*len(id_list)
  report = list(set(report))
  id = dict()
  ans_dict = dict()
  for i in range(len(id_list)):
    id[id_list[i]]=0
    ans_dict[id_list[i]]=0

  for i in report:
    a = i.split(' ')[0]
    b = i.split(' ')[1]
    id[b] +=1

  for i in range(len(report)):
    a = report[i].split(' ')[0]
    b = report[i].split(' ')[1]
    if id[b] >= k:
      ans_dict[a] += 1

  for i in range(len(id_list)):
    answer[i] += ans_dict[id_list[i]]

  return answer