[프로그래머스] 신고 결과 받기 (Python)

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

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

 

id_list = ["muzi", "frodo", "apeach", "neo"]
report = ["muzi frodo","apeach frodo","frodo neo","muzi neo","apeach muzi"]
k = 2

1. report = list(set(report)) 으로 report 중복 제거
2. n_dic (신고 당한 횟수 저장) 딕셔너리
3. dic (정지된 ID 개수 저장) 딕셔너리

신고 당한 횟수 딕셔너리에 저장 후 횟수가 k 이상이라면 dic 딕셔너리에 저장하고,
dic 딕셔너리의 value를 리스트 형태로 반환해주었다.

전체 코드

def solution(id_list, report, k):
    report = list(set(report))
    n_dic={}
    dic={}
    for i in id_list:
        n_dic[i]=0
        dic[i]=0
    for i in report:
        aname,bname = i.split()
        n_dic[bname]+=1
    for i in report :
        aname,bname = i.split()
        if n_dic[bname] >= k:
            dic[aname]+=1

    return list(dic.values())

print(solution(id_list, report, k))