Algorithm (PS)

[카카오] 신고 결과 받기 Python 풀이 - 해시 자료구조

minjiwoo 2022. 9. 29. 01:19
728x90

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

from collections import defaultdict
from collections import OrderedDict
def solution(id_list, report, k):
    answer = []
    id_dict = OrderedDict()
    count = defaultdict(int)
    
    for id in id_list:
        id_dict[id] = []
        
    for data in report:
        a, b = data.split()
        if b not in id_dict[a]:
            id_dict[a].append(b)
        else:
            continue
            
    for data in id_dict.values():
        if data:
            for id in data:
                count[id] += 1
    
    out = []
    for key, value in count.items():
        if value >= k:
            out.append(key)
    for key, value in id_dict.items():
        temp = 0 
        for id in out:
            if id in value:
                temp += 1
        answer.append(temp)
    return answer
728x90