Create Find Missing Observations.java

varunu28 · web-flow · commit 91581fb76b26 · 2021-10-03T08:53:47.000-05:00
diff --git a/Medium/Find Missing Observations.java b/Medium/Find Missing Observations.java
@@ -0,0 +1,17 @@
+class Solution {
+  public int[] missingRolls(int[] rolls, int mean, int n) {
+    int currSum = Arrays.stream(rolls).sum();
+    int missingSum = mean * (n + rolls.length) - currSum;
+    if (missingSum > n * 6 || missingSum < n) {
+      return new int[]{};
+    }
+    int[] ans = new int[n];
+    int part = missingSum / n;
+    int remainder = missingSum % n;
+    Arrays.fill(ans, part);
+    for (int i = 0; i < remainder; i++) {
+      ans[i]++;
+    }
+    return ans;
+  }
+}
