leetcode原题链接:组合总和
上一篇:HOT57-电话号码的字母组合
下一篇:HOT59-括号生成
题目描述
给你一个 无重复元素 的整数数组 candidates
和一个目标整数 target
,找出 candidates
中可以使数字和为目标数 target
的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。candidates
中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target
的不同组合数少于 150
个。
示例 1:
输入:candidates =[2,3,6,7],
target =7
输出:[[2,2,3],[7]] 解释: 2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。 7 也是一个候选, 7 = 7 。 仅有这两种组合。
示例 2:
输入: candidates = [2,3,5],
target = 8
输出: [[2,2,2,2],[2,3,3],[3,5]]
示例 3:
输入: candidates = [2],
target = 1
输出: []
提示:
1 <= candidates.length <= 30
2 <= candidates[i] <= 40
candidates
的所有元素 互不相同1 <= target <= 40
解题方法:回溯法。backtrack(candidates,target, i, tmp, results)表示对candidates[i]加入候选集对判断,如果满足加入条件(target>=candidates[i]),则继续调用backtrack(candidates,target-candidates[i], i, tmp, results),由于i可以重复选择,所以这里下一步还是调用i,而不是i+1。如果直接跳过i,则不需要将当前值i放入解中,直接调用backtrack(candidates,target, i+1, tmp, results)跳过解。
C++代码
#include <iostream>
#include <vector>
class Solution {
public:
std::vector<std::vector<int>> combinationSum(std::vector<int>& candidates, int target) {
std::vector<int> tmp;
std::vector<std::vector<int>> results;
backtrack(candidates, target, 0, tmp, results);
return results;
}
void backtrack(std::vector<int>& candidates,
const int& target,
int i,
std::vector<int>& tmp,
std::vector<std::vector<int>>& results) {
int n = candidates.size();
if (i == n) {
return;
}
if (target == 0) { //找到一个解
results.emplace_back(tmp);
return; //这里的return容易被落下
}
// 直接跳过第i个元素,这里跟下面选择第i个元素的顺序可以交换
backtrack(candidates, target, i + 1, tmp, results);
// 选择第i个元素
if (target - candidates[i] >= 0) {
tmp.emplace_back(candidates[i]);//选择第candidates
backtrack(candidates, target - candidates[i], i, tmp, results);//第i个元素可以重复选取
tmp.pop_back();
}
}
};