从零开始的CPP(33)多种终止条件的回溯

leetcode39

给你一个 无重复元素 的整数数组 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 3 6 7 都试一遍。不符合要求(sum>target)就弹栈(temp.pop_back),符合要求(sum=target)就把temp存入res再弹栈。为了防止重复搜索,设置一个k,保证只搜索索引>=当前索引的,不会搜索之前搜索过的

cpp 复制代码
class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
    vector<vector<int>> res = {};
    vector<int> temp = {};
    int sum = 0;
    int k = 0;
    helper(k,candidates, candidates[0], sum, temp, target, res);   
    return res;   
}
void helper(int k, vector<int>& candidates, int candidate, int& sum, vector<int>& temp, int target, vector<vector<int>>& res) {
    if (sum == target) {
        res.push_back(temp);
        sum = sum - temp[temp.size() - 1];
        temp.pop_back();
        return;
    }
    if (sum > target) {
        temp.pop_back();
        sum = sum- candidate;
        return;
    }
    for (int i = k; i < candidates.size(); i++) {
        temp.push_back(candidates[i]);
        sum = sum + candidates[i];
        helper(i,candidates, candidates[i],sum, temp, target, res);         
    }
    if (temp.empty()) return;        
        sum = sum - temp[temp.size() - 1];
        temp.pop_back();
        return;       
}
};
相关推荐
徐子童7 小时前
优选算法---链表
数据结构·算法·链表·面试题
如意.7597 小时前
从零开始的指针(3)
算法
CYH&JK7 小时前
数据结构---链式队列
数据结构
cwplh7 小时前
MX模拟赛总结
算法·动态规划
浅川.257 小时前
xtuoj 随机数
算法
shan&cen7 小时前
Day02 集合 | 30. 串联所有单词的子串、146. LRU 缓存、811. 子域名访问计数
java·数据结构·算法·缓存
阿方.9188 小时前
《树与二叉树详解:概念、结构及应用》
数据结构·二叉树··知识分享
NAGNIP8 小时前
大模型微调框架之TRL
算法
麦当_8 小时前
SwipeMultiContainer 滑动切换容器算法指南
前端·javascript·算法
橘子138 小时前
递归,搜索与回溯算法
算法