Create last-stone-weight-ii.cpp

Author: kamyu104

C++/last-stone-weight-ii.cpp

@@ -0,0 +1,22 @@
+// Time:  O(n * s), s is sum(stones)
+// Space: O(s)
+
+class Solution {
+public:
+    int lastStoneWeightII(vector<int>& stones) {
+        unordered_set<int> dp = {0};
+        const auto& S = accumulate(stones.cbegin(), stones.cend(), 0);
+        for (const auto& stone : stones) {
+            auto tmp(dp);
+            for (const auto& i : dp) {
+                tmp.emplace(i + stone);
+            }
+            dp = move(tmp);
+        };
+        auto result = numeric_limits<int>::max();
+        for (const auto& i : dp) {
+            result = min(result, abs(i - (S - i)));
+        }
+        return result;
+    }
+};