Leetcode 4.2

回溯

1.全排列

全排列
回溯算法的基本思想是:从一条路往前走,能进则进,不能进则退回来,换一条路再试。
在这里插入图片描述

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> tmp;

    vector<vector<int>> permute(vector<int>& nums) {
        vector<bool> used(nums.size(), false);
        dfs(nums, used);
        return ans;
    }

    void dfs(vector<int>& nums, vector<bool>& used) {
        if (tmp.size() == nums.size()) {
            ans.push_back(tmp);
            return;
        }
        for (int i = 0; i < nums.size(); i++) {
            if (used[i] == true) continue;
            tmp.push_back(nums[i]);
            used[i] = true;
            dfs(nums, used);
            tmp.pop_back();
            used[i] = false;
        }
    }

};

2.子集

子集
vector<vector> res;
vector track;

题解参考链接
①递归树
观察上图可得,选择列表里的数,都是选择路径(红色框)后面的数,比如[1]这条路径,他后面的选择列表只有 “2、3”,[2] 这条路径后面只有 “3” 这个选择,那么这个时候,就应该 使用一个参数 start,来标识当前的选择列表的起始位置。也就是标识每一层的状态,因此被形象的称为 “状态变量”.

//nums为题目中的给的数组
//path为路径结果,要把每一条 path 加入结果集
void backtrack(vector<int>nums,vector<int>&path,int start)

②找结束条件
此题非常特殊,所有路径都应该加入结果集,所以不存在结束条件。或者说当 start 参数越过数组边界的时候,程序就自己跳过下一层递归了,因此不需要手写结束条件,直接加入结果集

**res为结果集,是全局变量vector<vector<int>>res,到时候要返回的
res.push_back(path);//把每一条路径加入结果集

③找选择列表
在①中已经提到过了,子集问题的选择列表,是上一条选择路径之后的数,即

for(int i=start;i<nums.size();i++)

④判断是否需要剪枝
从递归树中看到,路径没有重复的,也没有不符合条件的,所以不需要剪枝

⑤做出选择(即for 循环里面的)

void backtrack(vector<int>nums,vector<int>&path,int start)
{
    for(int i=start;i<nums.size();i++)
    {
        path.push_back(nums[i]);//做出选择
        backtrack(nums,path,i+1);//递归进入下一层,注意i+1,标识下一个选择列表的开始位置,最重要的一步
    }
}

⑤撤销选择
整体的 backtrack 函数如下

C++
void backtrack(vector<int>nums,vector<int>&path,int start)
{
    res.push_back(path);
    for(int i=start;i<nums.size();i++)
    {
        path.push_back(nums[i]);//做出选择
        backtrack(nums,path,i+1);//递归进入下一层,注意i+1,标识下一个选择列表的开始位置,最重要的一步
        path.pop_back();//撤销选择
    }
}

代码

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> tmp;
    vector<vector<int>> subsets(vector<int>& nums) {
        dfs(nums, 0);
        return ans;
    }
    
    void dfs(vector<int>& nums, int start) {
        ans.push_back(tmp);
        for (int i = start; i < nums.size(); i++) {
            tmp.push_back(nums[i]);
            dfs(nums, i + 1);
            tmp.pop_back();
        }
    }
};

3.组合总和

组合总和
在这里插入图片描述
这里依然是定义两个全局变量,二维数组ans存放结果集,数组tmp存放符合条件的结果。(这两个变量可以作为函数参数传入)

首先是题目中给出的参数,集合candidates, 和目标值target。

本题还需要start Index来控制for循环的起始位置。

  • 递归终止条件

从叶子节点可以清晰看到,终止只有两种情况,sum大于target和sum等于target。
sum等于target的时候,需要收集结果。

  • 单层搜索的逻辑

单层for循环依然是从startIndex开始,搜索candidates集合。

注意本题元素为可重复选取的。

如何重复选取呢,看代码,注释部分:

for (int i = startIndex; i < candidates.size(); i++) {
    sum += candidates[i];
    path.push_back(candidates[i]);
    backtracking(candidates, target, sum, i); // 关键点:不用i+1了,表示可以重复读取当前的数
    sum -= candidates[i];   // 回溯
    path.pop_back();        // 回溯
}
class Solution {
public:
    vector<vector<int>> ans;
    vector<int> tmp;
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        dfs(candidates, target, 0, 0);
        return ans;
    }
    void dfs(vector<int>& candidates, int target, int sum, int start) {
        if (target < sum) {
            return;
        }
        if (target == sum) {
            ans.push_back(tmp);
            return;
        }
        for (int i = start; i < candidates.size(); i++) {
            tmp.push_back(candidates[i]);
            sum += candidates[i];
            dfs(candidates, target, sum, i); //i不需要+1,因为可以重复取数
            tmp.pop_back();
            sum -= candidates[i];
        }
    }
};
  • 5
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值