interviewbit: combinational sum2

Backtracking/CombinationSumII.cpp

@@ -1,44 +1,30 @@
-void make(vector<int>& A, int curr, int currSum, int sum, vector<int> temp, vector<vector<int> >& ans, map<vector<int>, bool>& m){
-    if(currSum > sum){
-        return;
-    }    
-    else if(currSum == sum){
-        if(m.find(temp) == m.end()){
-            m[temp] = true;
-            ans.push_back(temp);
-        }
+void back_track(vector<vector<int>> &ans, vector<int> &combination, vector<int>& candidates, int start, int target) {
+    if (target == 0) {
+        ans.push_back(combination);
         return;
     }
-
-    for(int i = curr; i < A.size(); i++){
-        vector<int> t(temp);
-        t.push_back(A[i]);
-        make(A, i+1, currSum+A[i], sum, t, ans, m);
+
+    for ( int i = start; i < candidates.size(); i++) {
+        if ((target - candidates[i]) < 0)
+            return;
+        if (candidates[i] <= target) {
+            combination.push_back(candidates[i]);
+            back_track(ans, combination, candidates, i + 1, target - candidates[i]);
+            combination.pop_back();  
+            while(i+1<candidates.size() && candidates[i]==candidates[i+1]) {
+                i++;
+            }
+        }
     }
 }
 
 vector<vector<int> > Solution::combinationSum(vector<int> &A, int B) {
-    // Do not write main() function.
-    // Do not read input, instead use the arguments to the function.
-    // Do not print the output, instead return values as specified
-    // Still have a doubt. Checkout www.interviewbit.com/pages/sample_codes/ for more details
-
-    vector<vector<int> > ans;
-
-    int n = A.size();
-
-    if(A.size() == 0){
-        return ans;
-    }
-
+	vector<vector<int>> ans;
+    vector<int> combination;
+    vector<int>::iterator ip;
+
     sort(A.begin(), A.end());
-    map<vector<int>, bool> m;
+    back_track(ans, combination, A, 0, B);
 
-    for(int i = 0; i < n; i++){
-        vector<int> temp;
-        temp.push_back(A[i]);
-        make(A, i+1, A[i], B, temp, ans, m);
-    }
-
-    return ans;
-}
+	return ans;
+}