代码随想录算法训练营第25天 | 491.递增子序列、46.全排列、47.全排列Ⅱ

代码随想录算法训练营第25天 | 491.递增子序列、46.全排列、47.全排列Ⅱ

---

491.递增子序列

题目链接：491.递增子序列

解题思路

本题抽象为树形结构如图：

回溯三部曲如下：

• 确定递归函数的参数和返回值：因为本题要求是子序列，一个元素不能重复使用，需要startIndex，另外需要两个数组记录单个结果和所有符合条件的数组集合
• 终止条件：不需要加终止条件，startIndex每次加1，不会无限递归
• 单层递归逻辑：同一父节点得同一层使用过的元素就不能再使用了，使用set对本层元素进行去重

代码实现

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums, int startIndex) {
        if (path.size() > 1) {
            result.push_back(path);
        }
        unordered_set<int> uset;
        for (int i = startIndex; i < nums.size(); i++) {
            if ((!path.empty() && nums[i] < path.back()) ||
                uset.find(nums[i]) != uset.end()) {
                continue;
            }
            uset.insert(nums[i]);
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
        }
    }

    vector<vector<int>> findSubsequences(vector<int>& nums) {
        result.clear();
        path.clear();
        backtracking(nums, 0);
        return result;
    }
};

题目总结

在单层递归时，uset 的作用是记录本层元素是否重复使用，新的一层都会重新定义 uset，uset 只负责递归过程中本层的逻辑。

46.全排列

题目链接：46.全排列

解题思路

使用回溯算法解题：

• 确定递归函数的参数与返回值：在处理排列问题时，不再使用 startIndex，但需要一个 used 数组，用于标记已经选择的元素
• 终止条件：到达叶子节点
• 确定单层递归的逻辑：每次从头开始搜索，used 数组用于记录此时 path 中哪些元素被使用了，一个排列中的元素只能使用一次

代码实现

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

    vector<vector<int>> permute(vector<int>& nums) {
        result.clear();
        path.clear();
        vector<bool> used(nums.size(), false);
        backtracking(nums, used);
        return result;
    }
};

题目总结

排列问题和组合问题的不同之处：

• 每层都是从0开始搜索，而不是从 startIndex 开始
• 需要 used 数组记录 path 中都存放了哪些元素

47.全排列Ⅱ

题目链接：47.全排列Ⅱ

解题思路

和上题的区别在于去重，之前组合问题中也有涉及

代码实现

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(vector<int>& nums, vector<bool>& used) {
        if (path.size() == nums.size()) {
            result.push_back(path);
            return;
        }

        for (int i = 0; i < nums.size(); i++) {
            if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
                continue;
            }
            if (used[i] == false) {
                used[i] = true;
                path.push_back(nums[i]);
                backtracking(nums, used);
                used[i] = false;
                path.pop_back();
            }
        }
    }

    vector<vector<int>> permuteUnique(vector<int>& nums) {
        path.clear();
        result.clear();
        sort(nums.begin(), nums.end());
        vector<bool> used(nums.size(), false);
        backtracking(nums, used);
        return result;
    }
};

题目总结

还是使用之前涉及到的去重方法。