Mar 31

Author: siddydutta

2025-03-March-LeetCoding-Challenge/Put Marbles in Bags.py

@@ -0,0 +1,25 @@
+from itertools import pairwise
+
+
+class Solution:
+    def putMarbles(self, weights: list[int], k: int) -> int:
+        pair_sum = [w1 + w2 for w1, w2 in pairwise(weights)]
+        pair_sum.sort()
+        diff = 0
+        for i in range(k - 1):
+            diff += pair_sum[len(weights) - 2 - i] - pair_sum[i]
+        return diff
+
+
+def main():
+    weights = [1, 3, 5, 1]
+    k = 2
+    assert Solution().putMarbles(weights, k) == 4
+
+    weights = [1, 3]
+    k = 2
+    assert Solution().putMarbles(weights, k) == 0
+
+
+if __name__ == '__main__':
+    main()

2025-03-March-LeetCoding-Challenge/README.md

@@ -34,12 +34,12 @@
 | March 28 | [2503. Maximum Number of Points From Grid Queries](https://leetcode.com/problems/maximum-number-of-points-from-grid-queries/) | Hard | Unsolved |
 | March 29 | [2818. Apply Operations to Maximize Score](https://leetcode.com/problems/apply-operations-to-maximize-score/) | Hard | Unsolved |
 | March 30 | [763. Partition Labels](https://leetcode.com/problems/partition-labels/) | Medium | Solved |
-| March 31 | []() |  |  |
+| March 31 | [2551. Put Marbles in Bags](https://leetcode.com/problems/put-marbles-in-bags/) | Hard | Solved |
 
 
 ## Summary
 | Level  | Problems | Solved | Unsolved |
 | ---    | --- | --- | --- |
 | Easy   | 6 | 6 | 0 |
 | Medium | 21 | 10 | 11 |
-| Hard   | 3 | 0 | 3 |
+| Hard   | 4 | 1 | 3 |