프로그래머스) 신고 결과 받기

https://programmers.co.kr/learn/courses/30/lessons/92334?language=python3 

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

 

내 풀이(맞음)

def solution(id_list, report, k):
    answer = []
    name_dic = {}
    for i in id_list:
        name_dic[i] = 0
        
    dic = {}
    for i in range(len(report)):
        t = report[i].split()
        if t[1] in dic:
            if t[0] in dic[t[1]]:
                continue
            else:
                dic[t[1]][0] += 1
                dic[t[1]].append(t[0])
        else:
            dic[t[1]] = [1, t[0]]
    for i in dic:
        if dic[i][0] >= k:
            for j in range(1, len(dic[i])):
                a = dic[i][j]
                name_dic[a] += 1
    for i in name_dic:
        answer.append(name_dic[i])
    return answer

name_dic에는 id_list를 기반으로 하여 각 사람의 신고 성공 횟수를 기록한다.

dic에서는 신고 당한 사람이 키값이고 벨류값은 집합으로해서 0번째에는 신고 당한 횟수를 넣고 그 뒤에는 신고한 사람의 이름을 넣어준다.

그리고 dic의 0번째에 있는 신고 당한 횟수가 k보다 크면 dic 1번째부터 끝까지 사람들에게 +1을 하여 메일이 보내졌다고 표시한다. 그래서 name_dic의 사람들에게 1씩 더해준다.