Create 416.Partition-Equal-Subset-Sum_dfs_v2.cpp

kushal-iitb · Feb 10, 2022 · 6e3ba1c · 6e3ba1c
1 parent 10a5112
commit 6e3ba1c
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/Dynamic_Programming/416.Partition-Equal-Subset-Sum/416.Partition-Equal-Subset-Sum_dfs_v2.cpp b/Dynamic_Programming/416.Partition-Equal-Subset-Sum/416.Partition-Equal-Subset-Sum_dfs_v2.cpp
@@ -0,0 +1,32 @@
+class Solution {
+    unordered_set<int>visited;
+public:
+    bool canPartition(vector<int>& nums) 
+    {
+        int sum = accumulate(nums.begin(), nums.end(), 0);
+        if (sum%2!=0) return false;
+        return dfs(nums, sum/2, 0, 0);        
+    }
+
+    bool dfs(vector<int>& nums, int targetSum, int i, int sum)
+    {
+        int hash = sum * 20005 + i;
+        if (visited.find(hash)!=visited.end())
+            return false;
+
+        if (sum== targetSum)
+            return true;
+
+        if (sum> targetSum)
+            return false;
+
+        if (i == nums.size())
+            return false;
+
+        if (dfs(nums, targetSum, i + 1, sum+ nums[i]) || dfs(nums, targetSum, i + 1, sum))
+            return true;
+
+        visited.insert(hash);
+        return false;        
+    }
+};