[A1/kyu] 키위쥬스 (Scala)

더 좋은 방법이 있을 것 같은데 더 공부를 해봐야 할 것 같다.

 

 

package org.kmdigit.quiz.q1 class KiwiJuiceEasy { def thePouring(capacities: Array[Int], bottles: Array[Int], fromId: Array[Int], toId: Array[Int]) : Array[Int] = { require(capacities.length >= 2 && capacities.length <= 50) require(capacities.length == bottles.length) require(fromId.length >= 1 && fromId.length <= 50) require(fromId.length == toId.length) for (i <- 0 to (fromId.length - 1)) { val fromBottle = bottles(fromId(i)) val toBottle = bottles(toId(i)) val sumBottle = fromBottle + toBottle if (sumBottle > capacities(toId(i))) { bottles(toId(i)) = capacities(toId(i)) bottles(fromId(i)) = sumBottle - capacities(toId(i)) } else { bottles(toId(i)) = sumBottle bottles(fromId(i)) = 0 } } bottles } }