LEETCODE-数组总和
题目
给你一个 无重复元素 的整数数组
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
代码
class Solution {private: vector<vector<int>> res; vector<int> cur_res; void dfs(vector<int>& candidates, int index, int target) { auto sum = accumulate(cur_res.begin(), cur_res.end(), 0); if(sum == target) { res.emplace_back(cur_res); return; }else if (sum > target) { return; } else { for (auto i = index; i <candidates.size(); ++i ) { cur_res.push_back(candidates[i]); dfs(candidates, i, target); cur_res.pop_back(); }
} }public: vector<vector<int>> combinationSum(vector<int>& candidates, int target) { dfs(candidates,0,target); return res; }};思路
深度优先搜索(DFS)Depth Fist Search
-
初始化:首先,定义两个全局变量
res和cur_res。res用于存储所有满足条件的组合,cur_res则用于在搜索过程中临时存储当前的组合。 -
深度优先搜索:然后,在函数
dfs中实现深度优先搜索。这个函数接收三个参数:候选数组candidates、当前考虑的候选数字的索引index和目标值target。-
在
dfs函数中,首先计算cur_res中所有数字的和sum。 -
如果
sum等于target,则找到了一个有效的组合,将cur_res添加到res中并返回。 -
如果
sum大于target,则当前的组合已经不可能满足条件,直接返回。 -
否则,从索引
index开始,遍历candidates中的每个数字。对于每个数字,将其添加到cur_res中,并递归地调用dfs函数,然后再将其从cur_res中移除。这一步是为了枚举所有可能的组合。
-
-
主函数:最后,在主函数
combinationSum中,调用dfs函数开始搜索,并返回res作为结果。