这题跟Permutation那题的区别就是在于有重复元素,比如说input={1,2,2}, output={{1,2,2}, {2,1,2}, {2,2,1}}。所以关键在于去重。我们可以把2,2看成2个不同的2,先排序,然后根据nums[i]==nums[i-1]就可以知道重复元素出来了。我们的原则是保证前面的2排在后面的2前面,避免后面的2排在前面的2的前面,这样就可以去重了。
注意:
1) 要先排序,不然没法知道是不是重复元素出来了。
2) if (visited[i]) continue; 这行还是适用,因为这里i=0…nums.size()-1,所以两个2的visited[]不一样!!!
3)去重的关键在于下面者一行
if (i > 0 && (nums[i] == nums[i-1]) && !visited[i - 1]) continue;
**如果nums[i]==nums[i-1]但nums[i-1]又没有被visited过,说明后面的2要被排在前面的2前面了,这种情况要绝对避免,**所以continue。
4) 与Subset II区别。组合类重复元素去重的条件是:
if ((i!=index) && (nums[i-1]==nums[i])) //去重
continue;
上面没用到visited,但有i!=index。
class Solution {
public:
/*
* @param : A list of integers
* @return: A list of unique permutations
*/
vector<vector<int>> permuteUnique(vector<int> &nums) {
vector<vector<int>> results;
vector<int> sol;
vector<bool> visited(nums.size(), false);
if (nums.empty()) {
results.push_back(vector<int>());
return results;
}
sort(nums.begin(), nums.end());
helper(nums, visited, sol, results);
return results;
}
void helper(vector<int> &nums, vector<bool> &visited, vector<int> &sol, vector<vector<int>> &results) {
if (sol.size() == nums.size()) {
results.push_back(sol);
return;
}
for (int i = 0; i < nums.size(); ++i) {
if (visited[i]) continue;
if (i > 0 && (nums[i] == nums[i-1]) && !visited[i - 1]) continue;
visited[i] = true;
sol.push_back(nums[i]);
helper(nums, visited, sol, results);
visited[i] = false;
sol.pop_back();
}
}
};
上面算法的时间复杂度是O(S*n^2)。其中S是具体有多少种排列(注意S<n! 因为有重复元素)。怎么理解呢?下面是网上的评论,我觉得讲的非常好。
假如一个list中有且只有n个1,没有其他元素,想要获得不重复的全排列,则在DFS的第一层有n个iteration,因为只有第一个1会深入下去,而其他的1(共有n-1个)都会被continue掉,不会进入下一层,只有第一个[1]会有子孙。到了第二层,我们想构建一个有两个元素的组合,则仅有一个[1, 1] 的branch会继续深入,其他的branch都无法传下子孙(剩余共有n-2个分支,第二层的分支是第一个1和后面n-1个1中选取一个1进行组合,除第一个branch之外其他n-2个branch都会被continue掉),以此类推,第一层有O(n)的时间复杂度,第二层有O(n-1)的时间复杂度,第三层为O(n-2),如此相加直到最后一层,因此总时间复杂度为O(n^2)。
这种情况下仅有一个全排列,但全排列的数量S增多的时候,时间复杂度也线性增加,因此是O(S*n^2)。