[Programmers] 다단계 칫솔 판매 (Python)

📖  문제

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

코딩테스트 연습 - 다단계 칫솔 판매

 

💻  코드

from collections import defaultdict


def divide_revenue(revenue, seller, recommend, amount):
    yours = amount // 10
    mine = amount - yours
    revenue[seller] += mine
    if yours > 0 and recommend[seller] != '-':
        divide_revenue(revenue, recommend[seller], recommend, yours)


def solution(enroll, referral, seller, amount):
    recommend = dict()
    for i in range(len(enroll)):
        recommend[enroll[i]] = referral[i]
    revenue = defaultdict(int)
    for i, sr in enumerate(seller):
        divide_revenue(revenue, sr, recommend, amount[i] * 100)
    return [revenue[e] for e in enroll]

 

🙌  한마디

문제 설명보고 트리(Tree)라 어려울 줄 알았는데 딕셔너리에 저장되어있어 수월했다.

재귀 방식으로 풀었다.