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]
.
如果还按照之前做Permuation第一题的思路,需要考虑重复元素的处理,下面的代码使用了unordered_set的方法,来判断元素是否已经遍历过,这样在swap之前多一个判断,就可以延续之前的回溯方法。
class Solution {
public:
void permuteUniqueHelper(int i, vector<int>& nums, vector<vector<int> >& result) {
if (i == nums.size()) { result.emplace_back(nums); return; }
unordered_set<int> st;
for (int j = i; j < nums.size(); ++j) {
if (st.count(nums[j]) == 1) { continue; }
st.insert(nums[j]);
std::swap(nums[i], nums[j]);
permuteUniqueHelper(i + 1, nums, result);
std::swap(nums[i], nums[j]);
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
vector<vector<int> > result;
permuteUniqueHelper(0, nums, result);
return result;
}
};
第二种方法是也是回溯的思路,但是在判断是否重复的元素的时候,它使用了一个技巧,第一回溯递归函数的调用用的是传值,而不是传参。第二判断重复元素的时候,由于在之前就进行过一次排序,所以交换后不用再交换回来,这样方便以后重复元素的判断,具体思路也还没有理清。。。
class Solution {
public:
void recursion(vector<int> num, int i, int j, vector<vector<int> > &res) {
if (i == j-1) {
res.push_back(num);
return;
}
for (int k = i; k < j; k++) {
if (i != k && num[i] == num[k]) continue;
swap(num[i], num[k]);
recursion(num, i+1, j, res);
}
}
vector<vector<int> > permuteUnique(vector<int> &num) {
sort(num.begin(), num.end());
vector<vector<int> >res;
recursion(num, 0, num.size(), res);
return res;
}
};