题目描述
给定一个可包含重复数字的序列,返回所有不重复的全排列。
示例:
输入: [1,1,2]
输出:
[
[1,1,2],
[1,2,1],
[2,1,1]
]
思路
基本思路:递归退出条件为遍历到最后一个数字时,向res中push_back一个vector<int>;每次递归前交换当前下标与index下标对应元素的值,递归后再把这两个值换回来,以达到全排列的目的。
这道题目跟剑指offer面试题38字符串的排列类似,但是要求res中元素不重复,需要在循环中加一个判断,即当i与index不等且这两个下标对应元素的值相等时直接continue,这是因为此时交换元素没有意义,反而会多出很多相同的结果。
但是:牛客网上与剑指offer对应的这道编程题 https://www.nowcoder.com/practice/fe6b651b66ae47d7acce78ffdd9a96c7?tpId=13&tqId=11180&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking 的测试用例应该是有问题的,比如没有考虑两次交换也可能等于一次交换甚至不交换的结果,如aabb的第二个a和第二个b交换为abab,也可以是aabb先变为abba,再变为abab,这些测试用例在leetcode上都有,是通过不了的,所以只能用以下两种方法:
1、先将res的类型设为 set<vector<int>>进行dfs,因为这个类型是自动判重的,用insert插入,最后在强制转为vector<vector<int>>或者写个循环依次复制入一个vector<vector<int>>作为结果。
2、在得到有重复元素的vector<vector<int>>后,先用sort排序,这样相同的元素就相邻了,再用迭代器遍历容器,将相同元素erase。这里有一个要注意的问题,erase一定要给一个返回值,要不会返回一个野指针,iter不知道指的是哪里会报错。另外,返回的iter自动指向下个元素,用for循环要慎重,很容易将iter加两次,也会造成结果错误。
代码(方法一)
class Solution {
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
set<vector<int>> res;
permute(nums, 0, res);
return vector<vector<int>> (res.begin(), res.end());
}
void permute(vector<int> &nums, int start, set<vector<int>> &res) {
if (start >= nums.size()){
res.insert(nums);
return;
}
for (int i = start; i < nums.size(); ++i) {
// if (i != start && nums[i] == nums[start])
// continue;
swap(nums[i], nums[start]);
permute(nums, start + 1, res);
swap(nums[i], nums[start]);
}
}
};
代码(方法二)
class Solution {
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
vector<vector<int>> res;
permute(nums, 0, res);
sort(res.begin(),res.end());
for(auto iter=res.begin();iter!=res.end();){
if(iter+1!=res.end() && *iter == *(iter+1))
iter = res.erase(iter);
else
iter++;
}
return res;
}
void permute(vector<int> &nums, int start, vector<vector<int>> &res) {
if (start == nums.size()-1){
res.push_back(nums);
return;
}
for (int i = start; i < nums.size(); ++i) {
if (i != start && nums[i] == nums[start])
continue;
swap(nums[i], nums[start]);
permute(nums, start + 1, res);
swap(nums[i], nums[start]);
}
}
};