Leetcode刷题---46

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

输入： [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> permute(vector& nums) {
        vector> res;
        backtrack(res, nums, 0,  (int)nums.size());
        return res;
    }
    void backtrack(vector>& res, vector& 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> permute(vector& nums) {
        vector> ans;
        backtracking(nums, 0, ans);
        return ans;
    }
    // 辅函数
    void backtracking(vector& nums, int level, vector>& 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]); 
        }
    }
};