代码随想录DAY32：回溯

最新推荐文章于 2024-07-22 04:47:17 发布

wertvoll

最新推荐文章于 2024-07-22 04:47:17 发布

阅读量39

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_45845722/article/details/129133604

版权

回溯

491. 递增子序列

题目描述

给你一个整数数组 nums ，找出并返回所有该数组中不同的递增子序列，递增子序列中至少有两个元素。你可以按任意顺序返回答案。

数组中可能含有重复元素，如出现两个整数相等，也可以视作递增序列的一种特殊情况。

思路

不能借助排序，借助 set 避免重复选择本层的相同元素；
加一个 if 判断选择大于等于当前元素的元素

代码

class Solution {
    //难点：不能先排序再选择
private:
    vector<vector<int>> res;
    vector<int> track;
    void backtrack(vector<int>& nums, int start)
    {
        if (track.size() >= 2)
        {
            res.push_back(track);
        }
        
        unordered_set<int> uset;
        for (int i = start; i < nums.size(); i++)
        {
            if (uset.find(nums[i]) != uset.end())
            {
                continue;
            }
            if (!track.empty() &&  nums[i] < track.back())
            {
                continue;
            }
            uset.insert(nums[i]);
            track.push_back(nums[i]);
            backtrack(nums, i + 1);
            track.pop_back();
        }

    }

public:
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        res.clear();
        track.clear();
        backtrack(nums, 0);
        return res;
    }
};

46.全排列

题目描述

给定一个不含重复数字的数组 nums ，返回其所有可能的全排列。你可以按任意顺序返回答案。

思路

排列：元素不重复，不可重复选择
借助 used

代码

class Solution {
    //排列：元素不重复，排列 借助used
private:
    vector<vector<int>> res;
    vector<int> track;
    void backtrack(vector<int>& nums, vector<bool>& used)
    {
        if (track.size() == nums.size())
        {
            res.push_back(track);
        }

        for (int i = 0; i < nums.size(); i++)
        {
            if (used[i])
            {
                continue;
            }
            used[i] = true;
            track.push_back(nums[i]);
            backtrack(nums,used);
            used[i] = false;
            track.pop_back();
        }
    }
public:
    vector<vector<int>> permute(vector<int>& nums) {
        res.clear();
        track.clear();
        vector<bool> used(nums.size(), false);
        backtrack(nums,used);
        return res;
    }
};

47.全排列II

题目描述

给定一个可包含重复数字的序列 nums ，按任意顺序返回所有不重复的全排列。

思路

排列：元素可重复，不可重复选择
避免相同元素产生相同排列，需要剪枝

代码

class Solution {
private:
    vector<vector<int>> res;
    vector<int> track;
    void backtrack(vector<int>& nums, vector<bool>& used)
    {
        if (track.size() == nums.size())
        {
            res.push_back(track);
            return;
        }

        for (int i = 0; i < nums.size(); i++)
        {
            if (used[i])
            {
                continue;
            }
            if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1])
            {
                continue;
            }
            used[i] = true;
            track.push_back(nums[i]);
            backtrack(nums, used);
            used[i] = false;
            track.pop_back();
        }
    }
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        res.clear();
        track.clear();
        vector<bool> used(nums.size(), false);
        sort(nums.begin(), nums.end());
        backtrack(nums, used);
        return res;
    }
};