[剑指offer] 数字排列

数字排列

题目

输入一组数字（可能包含重复数字），输出其所有的排列方式。

样例

输入：[1,   2,   3]

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

题解

由于本题会有重复数字，故解题思路如下

• 先将所有数从小到大排序，这样相同的数会排在一起
• 对于相同数，我们人为定序以避免重复计算：我们在 dfs 时记录一个额外的状态，记录上一个相同数存放的位置 start，我们在枚举当前数时，只枚举下标 start+1 以后的数字

class Solution {
public:
    int n;
    vector<vector<int>> ans;
    vector<int> line;

    void dfs(vector<int>& nums, int u, int start, int state) {
        if (u == n) {
            ans.push_back(line);
            return;
        }
        
        if (!u || nums[u] != nums[u - 1]) start = 0;
        for (int i = start; i < n; i ++)
            if (!(state >> i & 1)) {
                line[i] = nums[u];
                dfs(nums, u + 1, i + 1, state + (1 << i));
            }
    }

    vector<vector<int>> permutation(vector<int>& nums) {
        n = nums.size();
        line.resize(n);
        int state = 1 << n;
        sort(nums.begin(), nums.end());
        dfs(nums, 0, 0, state);
        return ans;
    }
};