LeetCode47 全排列II

链接: 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;
}
};