题意
给一个数组,数组内可能含有重复元素,求它的所有排列。
思路
首先,算法基于直接求数组的所有排列, 即Permutations。然后我们只需要考虑去重即可。
去重的思路同Combination Sum II:
- 先排序,然后dfs
- 对于两个相同的数,如果它之前的那个数没有使用,那么它也不能使用。
代码
int vis[10000 + 5];
class Solution {
private:
vector<vector<int>>ans;
public:
void dfs(vector<int>& a, vector<int>& num, int step) {
if (step == num.size()) {ans.push_back(a); return;}
for (int i = 0; i < num.size(); i++) {
if (vis[i]) continue;
if (i && num[i] == num[i - 1] && !vis[i - 1]) continue;
vis[i] = 1; a.push_back(num[i]);
dfs(a, num, step + 1);
vis[i] = 0; a.pop_back();
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
vector<int> a;
sort(nums.begin(), nums.end());
dfs(a, nums, 0);
return ans;
}
};