题目:
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(求全排列)