「代码随想录算法训练营」第二十二天 | 回溯算法 part4

491. 非递减子序列

题目链接:https://leetcode.cn/problems/non-decreasing-subsequences/
题目难度:中等
文章讲解:https://programmercarl.com/0491.递增子序列.html
视频讲解:https://www.bilibili.com/video/BV1EG4y1h78v/
题目状态:有思路,借助 ChatGPT 通过

思路:

在之前代码的基础上,添加一个set<vector<int>>类型的全局变量,用来跟踪已添加的子序列,来避免重复子序列的存储。

代码:

class Solution {
public:
    vector<vector<int>> res;
    vector<int> vec;
    set<vector<int>> seen;
    
    void backtracking(vector<int> &nums, int startIdx) {
        if(vec.size() >= 2) {
            if(seen.find(vec) == seen.end()) {
                res.push_back(vec);
                seen.insert(vec);
            }
        }
        for(int i = startIdx; i < nums.size(); ++i) {
            if(vec.empty() || nums[i] >= vec.back()) {
                vec.push_back(nums[i]);
                backtracking(nums, i + 1);
                vec.pop_back();
            }
        }
    }
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        backtracking(nums, 0);
        return res;
    }
};

46. 全排列

题目链接:https://leetcode.cn/problems/permutations/
题目难度:中等
文章讲解:https://programmercarl.com/0046.全排列.html
视频讲解:https://www.bilibili.com/video/BV19v4y1S79W/
题目状态:有思路,借助 ChatGPT 通过

思路:

定义一个全局变量vector<bool> isUsed,用来判断当前节点是否已经被回溯到了,若已经用过就直接跳过,之后使用回溯模板。

代码:

class Solution {
public:
    vector<vector<int>> res;
    vector<int> vec;
    vector<bool> isUsed;
    void backtracking(vector<int> &nums) {
        if(vec.size() == nums.size()) {
            res.push_back(vec);
            return;
        }
        for(int i = 0; i < nums.size(); ++i) {
            if(isUsed[i]) continue;
            vec.push_back(nums[i]);
            isUsed[i] = true;
            backtracking(nums);
            vec.pop_back();
            isUsed[i] = false;
        }
    }
    vector<vector<int>> permute(vector<int>& nums) {
        isUsed.resize(nums.size(), false);
        backtracking(nums);
        return res;
    }
};

47. 全排列II

题目链接:https://leetcode.cn/problems/permutations-ii/
题目难度:中等
文章讲解:https://programmercarl.com/0047.全排列II.html
视频讲解:https://www.bilibili.com/video/BV1R84y1i7Tm/
题目状态:有思路,通过

思路:

将上面两题结合一下,分别创建一个vector<bool>的全局变量来判断当前元素是否已经被使用,避免排列出来的单个结果中出现重复的元素;在定义一个set<vector<int>>的全局变量,来判断所有排列中是否出现了重复的排列方案。

代码:

class Solution {
public:
    vector<vector<int>> res;
    vector<int> vec;
    vector<bool> isUsed;
    set<vector<int>> seen;
    void backtracking(vector<int> &nums) {
        if(vec.size() == nums.size()) {
            if(seen.find(vec) == seen.end()) {
                res.push_back(vec);
                seen.insert(vec);
            }
        }
        for(int i = 0; i < nums.size(); ++i) {
            if(isUsed[i]) continue;
            vec.push_back(nums[i]);
            isUsed[i] = true;
            backtracking(nums);
            vec.pop_back();
            isUsed[i] = false;
        }
    }
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        isUsed.resize(nums.size(), false);
        backtracking(nums);
        return res;
    }
};

热门相关:军少夜宠:小甜妻,乖!   美漫之大冬兵   翻天   福晋有喜:四爷,宠上天!   视死如归魏君子