성우리뷰

파이썬 다단계칫솔 판매

두원공대88학번뚜뚜 2021. 12. 4. 17:00

// : 나누고 난 다음 소수점 버리고 난 정수 답

#pair을 <인물-대가리>로 구성한 배열

def solution(enroll, referral, seller, amount):
    dic = {}
    answer = [0 for i in range(len(enroll))]
    for i in range(len(enroll)):
        list = []
        list.append(referral[i])
        list.append(0)
        dic[enroll[i]] = list

    for i in range(len(amount)):
        name = seller[i]
        money = amount[i]*100
        #이름이 -가 아닐 때까지
        while name != '-':         
            if(money //10 == 0):
                dic[name][1] += money
                name = '-'
                break
                    
            #그 이름에 돈을 더 넣고, 나머지는 위로 올린다
            dic[name][1] += money - money//10
            money = money - (money - money//10)
            name = dic[name][0]                    

    for i in range(len(enroll)) :
        answer[i]=dic[enroll[i]][1]
        
    return answer