题目:
Given a collection of numbers that might contain duplicates, return all possible unique permutations.
For example,
[1,1,2]
have the following unique permutations:
[1,1,2]
, [1,2,1]
, and [2,1,1]
.
思路:这道题和Permutations的区别就是,输入的数字数组中包含重复的元素。如果我们对重复的元素不做处理的话,我们就会产生重复的结果。那么如何避免这种重复呢?按照我们前面处理这类问题的方法,首先对输入数组进行排序,然后对于重复的元素循环时跳过递归函数的调用,只对第一个未被使用的元素进行递归,这一次结果会出现在第一个的递归函数结果中,因为后面的重复元素将重复此次的递归结果,后面的结果被略过。如果第一个重复元素前面的元素还没在前面的结果(本次tmp)中(!used[i-1]), 那么我们不需要递归。比如[1,1,1,2], 如果第一个1在结果中(used[0] = true),我们下次循环时,我们判断到第二个1,才进行递归和push进tmp, 否则会导致重复。这一点很重要,需要思考下。总体来说,我们需要对重复元素和前面元素的使用情况进行判断。
AC Code:
class Solution {
public:
vector<vector<int> > permuteUnique(vector<int> &num) {
vector<vector<int> > ret;
if(num.size() == 0) return ret;
vector<int> tmp;
vector<bool> used(num.size(), false);
sort(num.begin(), num.end());
permuteUnique_helper(num, tmp, used, ret);
return ret;
}
private:
void permuteUnique_helper(vector<int>& num, vector<int> tmp, vector<bool> used, vector<vector<int> >& ret)
{
if(tmp.size() == num.size())
{
ret.push_back(tmp);
return;
}
for(int i = 0; i < num.size(); i++)
{
if(i > 0 && !used[i-1] && num[i-1] == num[i]) continue;
if(!used[i])
{
tmp.push_back(num[i]);
used[i] = true;
permuteUnique_helper(num, tmp, used, ret);
tmp.pop_back();
used[i] = false;
}
}
return;
}
};
这个解法带有一般性,放到
Permutations中也是正确的,所以在面试时遇到,我们可以直接实现这个代码,不要假设元素没有重复,可以和面试官进行讨论,但是一般我们都要考虑重复元素。
这道题,我们也可以用迭代的方法去做,可以用next permutation的算法,保证顺序,从最小的排列到最大的排列迭代下去,代码相对冗长,一般面试不会考到这道题迭代的方法,不过可以看下这篇博文:
Permutations, Permutations II(求全排列)