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

 

Problem

 

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

---

Solution

 

1. 판매자의 부모와 이익을 초기화해준다.

 

    var parentDic:[String:String] = [:]
    var profitDic:[String:Int] = [:]
    
    for i in 0..<enroll.count {
        profitDic[enroll[i]] = 0
        parentDic[enroll[i]] = referral[i]
    }

 

2. 루트 부모가 나올 때까지 이익에서 10%를 차감한다.

 

여기서 주의해야 할 점은 n/10을 했을 때 0인 경우엔 더 이상 진행해주면 안됩니다.(안그럼 시간초과남)

 

그러므로 while문의 조건을 가장 최상부모인 "-" 이거나 n/10이 0보다 클 때로 걸어줍니다.

 

for i in 0..<seller.count {
        var tenPercent:Int = amount[i]*100
        var parent:String = seller[i]
        while parent != "-" && tenPercent > 0 {
            profitDic[parent]! += tenPercent - tenPercent/10
            parent = parentDic[parent]!
            tenPercent = tenPercent/10
        }
    }

 

3. 이익을 enroll순으로 반환해준다.

 

return enroll.map{profitDic[$0]!}

---

Source Code