给定一个可包含重复数字的序列,返回所有不重复的全排列。
示例:
输入: [1,1,2] 输出: [ [1,1,2], [1,2,1], [2,1,1] ]解题思路:
DFS,桶排序。由于数组中可能包含相同数字,采用了排序+制桶,然后进行DFS。
例如: 排序后的数组为nums=[1,1,2,2,2,3,3,4,4,5,6,6,6],他对应的桶如下:
vector<pair<int,int>> = [(1,2),(2,3),(3,2),(4,2),(5,1),(6,3)];其中(first,second),first为数据,second为数量
对比没有相同数字的全排列而言,只需要避开下列操作,比如,我们在第i次选择中选择了data,那么在前i-1次选择相同的情况下,不在选择data。因此,我们选择数据的对象变成桶,DFS选择下一层访问数据时,选择下一个桶的first,前提是这个桶的second>0。
C++代码
class Solution { public: vector<vector<int>> permuteUnique(vector<int>& nums) { sort(nums.begin(), nums.end()); int i; size = nums.size(); if (size == 0) return res; data.push_back(pair<int, int>(nums[0], 0)); for (i = 1; i <= size; i++) { if (nums[i - 1] == data[data.size() - 1].first) { data[data.size() - 1].second++; } else data.push_back(pair<int, int>(nums[i - 1], 1)); } N = data.size(); vector<int> road; DFS(0, road); return res; } private: void DFS(int pos, vector<int> road) { if (int(road.size()) == size) { res.push_back(road); return; } for (int i = 0; i < N; i++) { if (data[i].second > 0) { data[i].second--; road.push_back(data[i].first); int len = road.size(); DFS(i + 1, road); road.erase(road.begin() + len - 1, road.end()); data[i].second++; } } } int size; int N; vector<pair<int,int>> data; vector<vector<int>> res; };