2023年11月11日

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

  1. 初始化:首先,定义两个全局变量 rescur_resres 用于存储所有满足条件的组合,cur_res 则用于在搜索过程中临时存储当前的组合。

  2. 深度优先搜索:然后,在函数 dfs 中实现深度优先搜索。这个函数接收三个参数:候选数组 candidates、当前考虑的候选数字的索引 index 和目标值 target

    • dfs 函数中,首先计算 cur_res 中所有数字的和 sum

    • 如果 sum 等于 target,则找到了一个有效的组合,将 cur_res 添加到 res 中并返回。

    • 如果 sum 大于 target,则当前的组合已经不可能满足条件,直接返回。

    • 否则,从索引 index 开始,遍历 candidates 中的每个数字。对于每个数字,将其添加到 cur_res 中,并递归地调用 dfs 函数,然后再将其从 cur_res 中移除。这一步是为了枚举所有可能的组合。

  3. 主函数:最后,在主函数 combinationSum 中,调用 dfs 函数开始搜索,并返回 res 作为结果。