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

문제

구현 문제였고, set과 map(dictionary)를 사용해서 문제를 풀었다.

먼저 이름들을 idx에 맞도록 dictionary에 매핑을 시켰고,

신고를 당한 사람을 key로 두고, value들을 신고한 사람을 set으로 둬서 문제를 해결하였다.

시간 복잡도는 report의 크기를 N으로 하면,

O(N)일 거 같다

 

코드

def solution(id_list, report, k):
    answer = [0 for i in range(len(id_list))]
    mail = {}
    nameToIdx = {}
    for idx, idl in enumerate(id_list):
        nameToIdx[idl] = idx
    for re in report:
        userId, reportedId = re.split(" ")
        if nameToIdx[reportedId] not in mail:
            mail[nameToIdx[reportedId]] = set()
        mail[nameToIdx[reportedId]].add(nameToIdx[userId])
    for key, v in mail.items():
        if len(v) >= k:
            for i in v:
                answer[i] += 1
    return answer