프로그래머스 - 다단계 칫솔 판매

문제

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

프로그래머스

 

 

풀이

딕셔너리로 판매자의 추천자를 저장한다. 이를 통해 추천자를 타고 올라가서 root까지 찾을 수 있다.

계산 결과를 저장할 때도 딕셔너리를 사용하면 편하게 저장할 수 있다.

딕셔너리에 저장한 계산 결과들은 valuse() 함수를 이용해 구하면 된다.

 

 

코드

python

def solution(enroll, referral, seller, amount):
    answer = {name: 0 for name in enroll}
    
    parent_dic = dict(zip(enroll, referral))
    
    for i in range(len(seller)):
        income = amount[i] * 100
        receiver = seller[i]
        
        while income > 0 and receiver != '-':
            answer[receiver] += income - income // 10
            income //= 10
            receiver = parent_dic[receiver]
        
    return list(answer.values())