46.全排列 (Medium)*

题目描述*

给定一个没有重复数字的序列，返回其所有可能的全排列。

示例*

输入: [1,2,3]

输出:

[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]

代码*

简单的方法使用 next_permutation 库函数，直接获取下一个全排列。

回溯算法，比较简单的思路就是用一个 visited 记录已添加的数字，每次都添加未访问过的数字，用 unoredered_set 记录。不过这样比较慢，进阶一点的做法是直接在 nums 上交换数据，会快一点。

visitednums

class Solution {
private:
    vector<vector<int>> res;
public:
    vector<vector<int>> permute(vector<int>& nums) {
        vector<int> path;
        unordered_set<int> visited;
        backtrack(nums, path, visited);
        return res;
    }
    void backtrack(vector<int>& nums, vector<int>& path, unordered_set<int>& visited) {
        if(path.size() == nums.size()) {
            res.push_back(path);
            return;
        }
        for(int i = 0; i < nums.size(); i++) {
            if(visited.count(nums[i])) {
                continue;
            }
            visited.insert(nums[i]);
            path.push_back(nums[i]);
            backtrack(nums, path, visited);
            path.pop_back();
            visited.erase(nums[i]);
        }
    }
};

class Solution {
private:
    vector<vector<int>> res;
public:
    vector<vector<int>> permute(vector<int>& nums) {
        vector<int> path;
        backtrack(nums, 0);
        return res;
    }
    void backtrack(vector<int>& nums, int i) {
        if(i == nums.size()) {
            res.push_back(nums);
            return;
        }
        for(int j = i; j < nums.size(); j++) {
            if(i != j) {
                swap(nums[i], nums[j]);
            }
            backtrack(nums, i + 1);
            if(i != j) {
                swap(nums[i], nums[j]);
            }
        }
    }
};

最后更新: July 23, 2022