Create 2028-find-missing-observations.kt

Author: a93a

kotlin/2028-find-missing-observations.kt

@@ -0,0 +1,20 @@
+class Solution {
+    fun missingRolls(rolls: IntArray, mean: Int, _n: Int): IntArray {
+        val m = rolls.size
+        var n = _n
+        var nTotal = (mean * (n + m)) - (rolls.sum() ?: 0)
+
+        if (nTotal < n || nTotal > n * 6)
+            return IntArray (0)
+        
+        var res = LinkedList<Int>()
+        while (nTotal > 0) {
+            val dice = minOf(nTotal - n + 1, 6)
+            res.add(dice)
+            nTotal -= dice
+            n--
+        }
+
+        return res.toIntArray()
+    }
+}