Create 1049-last-stone-weight-ii.py

Author: neetcode-gh

python/1049-last-stone-weight-ii.py

@@ -0,0 +1,18 @@
+class Solution:
+    def lastStoneWeightII(self, stones: List[int]) -> int:
+        # Memoization
+        stoneSum = sum(stones)
+        target = ceil(stoneSum / 2)
+
+        def dfs(i, total):
+            if total >= target or i == len(stones):
+                return abs(total - (stoneSum - total))
+            if (i, total) in dp:
+                return dp[(i, total)]
+
+            dp[(i, total)] = min(dfs(i + 1, total),
+                                 dfs(i + 1, total + stones[i]))
+            return dp[(i, total)]
+
+        dp = {}
+        return dfs(0, 0)