[LeetCode][Kotlin] 799. Champagne Tower

799. Champagne Tower

We stack glasses in a pyramid, where the first row has 1 glass, the second row has 2 glasses, and so on until the 100th row.  Each glass holds one cup of champagne.

 

Then, some champagne is poured into the first glass at the top.  When the topmost glass is full, any excess liquid poured will fall equally to the glass immediately to the left and right of it.  When those glasses become full, any excess champagne will fall equally to the left and right of those glasses, and so on.  (A glass at the bottom row has its excess champagne fall on the floor.)

 

For example, after one cup of champagne is poured, the top most glass is full.  After two cups of champagne are poured, the two glasses on the second row are half full.  After three cups of champagne are poured, those two cups become full - there are 3 full glasses total now.  After four cups of champagne are poured, the third row has the middle glass half full, and the two outside glasses are a quarter full, as pictured below.

 

Now after pouring some non-negative integer cups of champagne, return how full the jth glass in the ith row is (both i and j are 0-indexed.)

우리는 피라미드 형태로 유리잔을 쌓습니다. 첫 번째 줄에는 유리잔 1개가 있고, 두 번째 줄에는 유리잔 2개가 있으며 이런 방식으로 100번째 행까지 계속됩니다. 각 잔에는 샴페인 한 잔이 담겨 있습니다. 

그런 다음 맨 위에 있는 첫 번째 잔에 샴페인을 붓습니다. 맨 위 유리잔이 가득 차면, 부은 여분의 액체는 유리잔 바로 왼쪽과 오른쪽에 균등하게 떨어집니다. 잔이 가득 차면 남은 샴페인이 잔의 왼쪽과 오른쪽으로 동일하게 떨어집니다. (맨 아래 줄의 유리잔에는 남은 샴페인이 바닥에 떨어졌습니다.) 

예를 들어, 샴페인 한 잔을 부은 후 가장 윗부분의 잔이 가득 찼습니다.  두 잔의 샴페인을 부은 후 두 번째 줄의 두 잔이 반쯤 찼습니다. 세 잔의 샴페인을 부으면 그 두 잔이 가득 차게 됩니다. 이제 총 3잔의 잔이 가득 찼습니다. 샴페인 4잔을 부은 후, 세 번째 줄은 아래 그림과 같이 가운데 잔이 반이 차고, 바깥쪽 잔 2개가 1/4이 찼습니다.

이제 음수가 아닌 정수 컵에 샴페인을 부은 후 i번째 행의 j번째 잔이 얼마나 가득 차 있는지 반환합니다(i와 j는 모두 0 인덱스입니다.)

 

Example 1:

Input: poured = 1, query_row = 1, query_glass = 1

Output: 0.00000

Explanation: We poured 1 cup of champange to the top glass of the tower (which is indexed as (0, 0)). There will be no excess liquid so all the glasses under the top glass will remain empty.

 

Example 2:

Input: poured = 2, query_row = 1, query_glass = 1

Output: 0.50000

Explanation: We poured 2 cups of champange to the top glass of the tower (which is indexed as (0, 0)). There is one cup of excess liquid. The glass indexed as (1, 0) and the glass indexed as (1, 1) will share the excess liquid equally, and each will get half cup of champange.

 

Example 3:

Input: poured = 100000009, query_row = 33, query_glass = 17

Output: 1.00000

 

Constraints:

• 0 <= poured <= 10^9
• 0 <= query_glass <= query_row < 100

---

코드

• pour = 현재 각 잔에 채워진 샴페인의 양을 저장한 리스트
• 다음 행의 잔 개수는 이전 행의 잔 개수 보다 하나 더 많다.
• 이전 행의 각 잔을 순회하면서 넘치는 샴페인의 양을 계산한다.
• 샴페인이 아래 잔으로 넘치려면, 이전 행의 잔에 남아있는 샴페인의 양이 1 보다 커야 한다.

class Solution {
    fun champagneTower(poured: Int, queryRow: Int, queryGlass: Int): Double {
        var pour = mutableListOf(poured.toDouble())

        for (i in 1..queryRow) {
            val arr = MutableList<Double>(i + 1) { 0.0 }
            for (j in 0 until i) {
                val extra = pour[j] - 1
                if (extra > 0) {
                    arr[j] += extra * 0.5
                    arr[j + 1] += extra * 0.5
                }
            }
            pour = arr
        }

        return minOf(1.0, pour[queryGlass])
    }
}