Leetcode刷题---46

最新推荐文章于 2024-11-05 21:58:11 发布

好久不见0327

最新推荐文章于 2024-11-05 21:58:11 发布

阅读量123

点赞数

分类专栏： Leetcode 文章标签：回溯算法数组排列 C++ 时间复杂度空间复杂度

本文链接：https://blog.csdn.net/m0_50394077/article/details/124890823

版权

Leetcode 专栏收录该内容

38 篇文章 0 订阅

订阅专栏

题目：输入是一个一维整数数组，输出是一个二维数组，表示输入数组的所有排列方式。

输入： [1,2,3]

输出：[[1,2,3], [1,3,2], [2,1,3], [2,3,1], [3,1,2], [3,2,1]]

方法一：回溯一

class Solution {
public:
    vector<vector<int>> permute(vector<int>& nums) {
        vector<vector<int>> res;
        backtrack(res, nums, 0,  (int)nums.size());
        return res;
    }
    void backtrack(vector<vector<int>>& res, vector<int>& output, int first, int len) {
        // 所有数都填完了
        if (first == len) {
            res.emplace_back(output);
            return;
        }
        for (int i = first; i < len; ++i) {
            // 动态维护数组
            swap(output[i], output[first]);
            // 继续递归填下一个数
            backtrack(res, output, first + 1, len);
            // 撤销操作
            swap(output[i], output[first]);
        }
    }
};

复杂度分析

时间复杂度：O(n x n!)

空间复杂度：O(n)

方法二：回溯二

class Solution {
public:
    // 主函数
    vector<vector<int>> permute(vector<int>& nums) {
        vector<vector<int>> ans;
        backtracking(nums, 0, ans);
        return ans;
    }
    // 辅函数
    void backtracking(vector<int>& nums, int level, vector<vector<int>>& ans) {
        if (level == nums.size() - 1) {
            ans.push_back(nums);
            return;
        }
        for (int i = level; i < nums.size(); i++) {
            swap(nums[i], nums[level]);             // 修改当前节点状态
            backtracking(nums, level + 1, ans);     // 递归子节点
            swap(nums[i], nums[level]); 
        }
    }
};