PROGRAMMERS 77486: 다단계 칫솔 판매

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

---

Python

class Node:
    def __init__(self, name):
        self.name = name
        self.amount = 0
        self.parent = None
 
    def __str__(self):
        return self.name + " : " + str(self.amount)
 
 
def traceNode(person: Node, amount):
    if amount < 1:  # 가격이 1원보다 작으면 분배 할 돈이 없으므로 바로 종료
        return
    if person.name == "center":  # 부모라면
        person.amount += amount  # 부모에게 돈을 전부 주고 종료
    else:
        my_amount = amount - (amount // 10)  # 자기 수익금 계산
        person.amount += my_amount  # 자기 수익금을 더해줌
 
        traceNode(person.parent, amount - my_amount)  # 자기 수익금을 뺀 부모를 다시 추적
 
 
def solution(enroll, referral, seller, amount):
    people = dict()
 
    people['center'] = Node('center')
    for name in enroll:
        people[name] = Node(name)  # 미리 사람들을 정의하고 dict에 바로 찾을 수 있게 넣어줌
 
    for child, parent in zip(enroll, referral): # (enroll[0], referral[0]) ... (enroll[n], referral[n]) 이 됨
        if parent == "-":  # 만약 부모가 없다면
            people[child].parent = people["center"]  # 민호에게 연결
        else:
            people[child].parent = people[parent]  # 부모에게 연결
 
    for seller, amount in zip(seller, amount):
        amount *= 100  # 칫솔 이익금은 한개당 100원
        traceNode(people[seller], amount)  # 부모를 추적하여 수익을 분배함
    return [people[person].amount for person in enroll]  # enroll 순서대로 수익금을 반환

FeedBack

1.  

---

 

 

 

cjw.git@gmail.com