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

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

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

1. 문제이해

 - 부모 - 자식 노드 연결 문제

 

2. 코드 작성

def solution(enroll, referral, seller, amount):
    answer = [0 for _ in range(len(enroll))]
    enroll_dict = dict()
    
    i = 0
    for en, re in zip(enroll, referral):
        enroll_dict[en] = (re, i)
        i += 1

    for sell, cnt in zip(seller, amount):
        money = cnt * 100

        while enroll_dict[sell][0] != '-' and money > 0:
            answer[enroll_dict[sell][1]] += money - (money // 10)
            money = money // 10
            sell = enroll_dict[sell][0]

        answer[enroll_dict[sell][1]] += money - (money // 10)

    return answer

 

3. 풀이간 체크사항

 1) List로 index 를 찾아 넣는 것 보다 dictionary 구조를 활용하는 것 => 시간초과 해결