LeetCode47 全排列II

发表于 | 分类于 |
字数统计: 282 | 阅读时长 ≈ 1

链接: https://leetcode.cn/problems/permutations-ii/

题意

给定一个可包含重复数字的序列 nums ，按任意顺序 返回所有不重复的全排列。

解法

用dfs+回溯解全排列问题
不同的地方在于序列中存在重复的元素
所以如果使用swap写法的话很麻烦（写不出来）
所以还是考虑使用vis进行append排列
通过定义prev函数维护上一次添加的数 如果是一样的话就不进行新一层的dfs

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class Solution {
public:
    vector<vector<int>> ans;
    vector<int> vis;
    void dfs(vector<int>& nums, int loc, vector<int>& cur) {
        // cout << loc << endl;
        if (loc == nums.size()) {
            ans.push_back(cur);
            return;
        }
        int prev = 101;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == prev) {
                // cout << nums[i] << ' ' << prev << endl;
                continue;
            }
            if (!vis[i]) {
                prev = nums[i];
                vis[i] = 1;
                cur.push_back(nums[i]);
                dfs(nums, loc + 1, cur);
                cur.pop_back();
                vis[i] = 0;
            }
        }
    }
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vis.resize(nums.size());
        sort(nums.begin(), nums.end());
        vector<int> cur;
        dfs(nums, 0, cur);
        return ans;
    }
};