首页 > 代码库 > 【LeetCode】Combination Sum II (2 solutions)
【LeetCode】Combination Sum II (2 solutions)
Combination Sum II
Given a collection of candidate numbers (C) and a target number (T), find all unique combinations in C where the candidate numbers sums to T.
Each number in C may only be used once in the combination.
Note:
- All numbers (including target) will be positive integers.
- Elements in a combination (a1, a2, … , ak) must be in non-descending order. (ie, a1 ≤ a2 ≤ … ≤ ak).
- The solution set must not contain duplicate combinations.
For example, given candidate set 10,1,2,7,6,1,5
and target 8
,
A solution set is: [1, 7]
[1, 2, 5]
[2, 6]
[1, 1, 6]
解法一:
最直接的想法就是把所有子集都列出来,然后逐个计算和是否为target
但是考虑到空间复杂度,10个数的num数组就有210个子集,因此必须进行“剪枝”,去掉不可能的子集。
先对num进行排序。
在遍历子集的过程中:
(1)单个元素大于target,则后续元素无需扫描了,直接返回结果。
(2)单个子集元素和大于target,则不用加入当前的子集容器了。
(3)单个子集元素和等于target,加入结果数组。
class Solution {public: vector<vector<int> > combinationSum2(vector<int> &num, int target) { vector<vector<int> > result; vector<vector<int> > subsets; vector<int> empty; subsets.push_back(empty); sort(num.begin(), num.end()); for(int i = 0; i < num.size();) {//for each number int count = 0; int cur = num[i]; if(cur > target) //end return result; while(i < num.size() && num[i] == cur) {//repeat count i ++; count ++; } int size = subsets.size(); //orinigal size instead of calling size() function for(int j = 0; j < size; j ++) { vector<int> sub = subsets[j]; int tempCount = count; while(tempCount --) { sub.push_back(cur); int sum = accumulate(sub.begin(), sub.end(), 0); if(sum == target) { result.push_back(sub); subsets.push_back(sub); } else if(sum < target) subsets.push_back(sub); } } } return result; }};
解法二:递归回溯
需要注意的是:
1、在同一层递归树中,如果某元素已经处理并进入下一层递归,那么与该元素相同的值就应该跳过。否则将出现重复。
例如:1,1,2,3
如果第一个1已经处理并进入下一层递归1,2,3
那么第二个1就应该跳过,因为后续所有情况都已经被覆盖掉。
2、相同元素第一个进入下一层递归,而不是任意一个
例如:1,1,2,3
如果第一个1已经处理并进入下一层递归1,2,3,那么两个1是可以同时成为可行解的
而如果选择的是第二个1并进入下一层递归2,3,那么不会出现两个1的解了。
class Solution {public: vector<vector<int> > combinationSum2(vector<int> &num, int target) { vector<vector<int> > result; vector<int> cur; sort(num.begin(), num.end()); Helper(num, result, cur, 0, target); return result; } void Helper(vector<int> &num, vector<vector<int> > &result, vector<int> cur, int ind, int target) { int last; //first element for(int i = ind; i < num.size(); i ++) { if(i == ind) { cur.push_back(num[i]); int sum = accumulate(cur.begin(), cur.end(), 0); if(sum == target) { result.push_back(cur); return; } else if(sum < target) Helper(num, result, cur, i+1, target); else return; cur.pop_back(); last = num[i]; //start from last has been tried } else { //start from different element while(i < num.size() && num[i] == last) i ++; if(i == num.size()) return; if(num[i] > target) return; cur.push_back(num[i]); int sum = accumulate(cur.begin(), cur.end(), 0); if(sum == target) { result.push_back(cur); return; } else if(sum < target) Helper(num, result, cur, i+1, target); else return; cur.pop_back(); last = num[i]; //start from last has been tried } } }};
【LeetCode】Combination Sum II (2 solutions)