LeetCode OJ:Permutations II(排列II)
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]
.
依旧是dfs,不过这次要注意的是排列不允许有重复的vector存在,那么在排列之前应该先排序,然后每次检查的时候,如果前面一个是相同的元素而且已经用过了那么这个元素才可以用,如果不相同的元素那么直接dfs就可以用了,代码如下:
1 class Solution { 2 public: 3 vector<vector<int>> permuteUnique(vector<int>& nums) { 4 memset(cdds, 0, sizeof(cdds)); 5 memset(mark, 0, sizeof(mark)); 6 sort(nums.begin(), nums.end()); 7 dfs(0, nums.size(), nums); 8 return ret; 9 } 10 11 void dfs(int dep, int maxDep, vector<int> & nums) 12 { 13 if(dep == maxDep){ 14 tmp.clear(); 15 for(int i = 0; i < maxDep; ++i) 16 tmp.push_back(cdds[i]); 17 ret.push_back(tmp); 18 } 19 for(int i = 0; i < maxDep; ++i){ 20 if(!mark[i]){ 21 if(i != 0 && nums[i] == nums[i - 1] && !mark[i - 1]) 22 continue; 23 mark[i] = true; 24 cdds[dep] = nums[i]; 25 dfs(dep + 1, maxDep, nums); 26 mark[i] = false; 27 } 28 } 29 } 30 private: 31 int cdds[100]; 32 bool mark[100]; 33 vector<int> tmp; 34 vector<vector<int>> ret; 35 };
做出这种限制的原因就是使得相同的数只存在一种排列顺序,这样就不会出现重复的组合这种情况了。