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

사용 언어 - Python3

문제 - 신고 결과 받기

프로그래머스

 

정답

defaultdict 이용한 딕셔너리 구현 (정답 맞춘 여부 X)

user = {신고한 사람 : 신고 받은 사람들}

cnt = {신고 받은 사람 : 신고 받은 횟수}

딕셔너리로 저장

id_list 별로 한개씩 확인,

user 딕셔너리에서 id 확인해 신고받은 사람들의 신고횟수가 k이상인 경우, result += 1

 

# defaultdict = dict 입력형식 지정 set or int
from collections import defaultdict

def solution(id_list,report,k):
    answer = []
    
    report = list(set(report)) #report 중복제거
    user = defaultdict(set) #신고한 id : 신고받은 id
    cnt = defaultdict(int) #신고한 사람별 신고 횟수
    
    for r in report:
        a,b = r.split() #신고 한 사람, 신고 받은 사람
        user[a].add(b) # {신고한 사람 : 신고 받은 사람들}
        cnt[b] += 1 # {신고받은 사람 : 신고 받은 횟수}
        
    for i in id_list:
        result = 0
        for u in user[i]: #id별
            if cnt[u] >= k: #id별 신고횟수 조사
                result += 1
        answer.append(result)
        
    return answer

 

 

레퍼런스

• from collections import defaultdict

[파이썬 기초] 유사 딕셔너리 defaultdict() 활용법

• 정답 깃허브

GitHub - yyeongeun/codingtest: 코딩테스트 공부