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

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

프로그래머스

---

def solution(id_list, report, k):
    answer = []
    report = list(set(report))
    
    dict_id_mail = {name : 0 for name in id_list}
    dict_id_report= {name : [] for name in id_list}
    
    for i in report:
        person1 = i.split()[0]
        person2 = i.split()[1]
        dict_id_report[person2].append(person1)
        
    for i in dict_id_report:
        if len(dict_id_report[i]) >= k:
            for j in dict_id_report[i]:
                dict_id_mail[j] += 1
    
    answer =  list(dict_id_mail.values())        
    return answer

---

좋아요 많은 코드

def solution(id_list, report, k):
    answer = [0] * len(id_list)    
    reports = {x : 0 for x in id_list}

    for r in set(report):
        reports[r.split()[1]] += 1

    for r in set(report):
        if reports[r.split()[1]] >= k:
            answer[id_list.index(r.split()[0])] += 1

    return answer

list.index 를 사용하여 역추적