프로그래머스 [Lv. 3] 다단계 칫솔 판매 {언어 : Python} [다시 풀어 보기]

문제 링크

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

프로그래머스

정답 코드

def solution(enroll, referral, seller, amount):
    lenE = len(enroll)
    answer = [0 for _ in range(lenE)]
    
    erDict = dict()
    erDict["-"] = -1  # 센터는 인덱스 -1로 표시
    for i in range(lenE):
        erDict[enroll[i]] = i
        
    parents = [-1 for _ in range(lenE)]
    for i in range(lenE):
        if referral[i] != "-":
            parents[i] = erDict[referral[i]]
    
    for i in range(len(seller)):
        current = erDict[seller[i]]
        money = amount[i] * 100
        
        while current != -1:
            keep = money - money // 10
            answer[current] += keep
            money //= 10
            current = parents[current]
            if money < 1:
                break
    
    return answer

풀이 방법

1. 해당 이름의 위치를 erDict에 저장한다.
2. 현재 위치의 부모 위치를 node형식으로 저장한다.
3. 판매자부터 거슬러 올라간다.
   1. 돈을 남겨가면서 올라간다.
   2. 1보다 작을 경우 상납할 돈이 없으므로 종료한다.
4. answer[i]에 더해진 값들이 상납된 금액의 합이다.

느낀점

• 잘 풀긴 했는데 마무리가 안좋아서 다시 풀어보는 것이 좋을 것 같다.