组合总和 II
组合总和 II
问题描述
给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。
说明:
所有数字(包括目标数)都是正整数。
解集不能包含重复的组合。
示例 1:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-ii
思路
本题分为两步
简单的回溯算法
对未完全剪枝的ret进行去重。
去重的方法有两种,效率都比较低
sort(ret.begin(), ret.end())
;ret.erase(unique(ret.begin(), ret.end()), ret.end())
set<vector<int>> sets(ret.begin(), ret.end())
;ret=vector<vector<int>>(sets.begin(),sets.end())
题解
class Solution {
public:
vector<int> tmp;
vector<vector<int>> ret;
int n;
vector<vector<int>> combinationSum2(vector<int> &candidates, int target) {
n = candidates.size();
sort(candidates.begin(), candidates.end());
dfs(candidates, 0, 0, target);
sort(ret.begin(), ret.end());
ret.erase(unique(ret.begin(), ret.end()), ret.end());
return ret;
}
void dfs(vector<int> vec, int sum, int location, int target) {
if (sum >= target || location == n) {
if (sum == target) {
ret.push_back(tmp);
}
return;
}
for (int i = location; i < n; i++) {
tmp.push_back(vec[i]);
sum += vec[i];
dfs(vec, sum, i + 1, target);
sum = sum - vec[i];
tmp.pop_back();
}
}
};