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

사용 개념

from collections import defaultdict

 

defaultdict란? 

 

 

정답 코드 :

from collections import defaultdict 
def solution(id_list, report, k):
    answer = []
    
    report = list(set(report))
    user = defaultdict(set)
    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]:
            if cnt[u] >= k:
                result += 1
        answer.append(result)
    return answer

 

내 코드 : 

def solution(id_list, report, k):
    answer = [0]*len(id_list)
    
    report = list(set(report))
    rlist = [0]*len(report)
    tmp = []
    
    for i in range(len(report)):
        x,y = map(str, report[i].split())
        rlist[i] = [x,y]
        tmp.append(y)
    
    for i in range(len(id_list)):
        if tmp.count(id_list[i]) >= k:
            for j in range(len(rlist)):
                if id_list[i] == rlist[j][1]:
                    answer[id_list.index(rlist[j][0])] += 1
    
    
    return answer