[프로그래머스 level3 다단계 칫솔 판매] 딕셔너리 활용하기

해결해야하는 문제는 다음과 같았다.

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

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

 

 

 

 

이러한 아이디어를 바탕으로 작성한 코드는 다음과 같다.

def solution(enroll, referral, seller, amount):
    answer = []
    dicts = {}
    tree_dicts = {}
    for i in range(len(enroll)):
        tree_dicts[enroll[i]] = referral[i]
        dicts[enroll[i]] = 0

    for i in range(len(seller)) :
        cost = amount[i]*100
        node = seller[i]
        while node != '-':
            parent = tree_dicts[node]
            if cost*0.1 < 1:
                dicts[node] += cost
                break
            dicts[node] += (cost-(cost//10))
            cost //= 10
            node = parent


    for name in enroll :
        answer.append(int(dicts[name]))
    return answer