2551_Put_Marbel_in_Bags.java

Author: vishwajeetmishra4

JAVA/2551_Put_Marbel_in_Bags.java

@@ -0,0 +1,24 @@
+// Problem Number: 2551
+
+// Put Marbels in Bags.
+
+class Solution {
+  public long putMarbles(int[] weights, int k) {
+    
+    int[] arr = new int[weights.length - 1]; 
+    long mn = 0;
+    long mx = 0;
+
+    for (int i = 0; i < arr.length; ++i)
+      arr[i] = weights[i] + weights[i + 1];
+
+    Arrays.sort(arr);
+
+    for (int i = 0; i < k - 1; ++i) {
+      mn += arr[i];
+      mx += arr[arr.length - 1 - i];
+    }
+
+    return mx - mn;
+  }
+}