题目
给定一个可包含重复数字的序列 nums
,按任意顺序 返回所有不重复的全排列。
思路
和全排列异曲同工,去重操作和非递减子序列有些相似
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
void permuteUniqueAux(vector<int>& nums, bool* used_elem) {
unordered_set<int> repeat_elem;
for(int i = 0; i < nums.size(); ++i)
{
if(!used_elem[i] && find(repeat_elem.begin(), repeat_elem.end(), nums[i]) != repeat_elem.end())
{
path.emplace_back(nums[i]);
if(path.size() == nums.size()) res.emplace_back(path);
used_elem[i] = true;
repeat_elem.insert(nums[i]);
permuteUniqueAux(nums, used_elem);
used_elem[i] = false;
path.pop_back();
}
else continue;
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
bool used_elem[nums.size()] = {false};
permuteUniqueAux(nums, );
return res;
}
};
看完代码随想录,发现在树枝上或树层上去重都是可以的
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking (vector<int>& nums, vector<bool>& used) {
// 此时说明找到了一组
if (path.size() == nums.size()) {
result.push_back(path);
return;
}
for (int i = 0; i < nums.size(); i++) {
// used[i - 1] == true,说明同一树枝nums[i - 1]使用过
// used[i - 1] == false,说明同一树层nums[i - 1]使用过
// 如果同一树层nums[i - 1]使用过则直接跳过
if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
continue;
}
if (used[i] == false) {
used[i] = true;
path.push_back(nums[i]);
backtracking(nums, used);
path.pop_back();
used[i] = false;
}
}
}
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
result.clear();
path.clear();
sort(nums.begin(), nums.end()); // 排序
vector<bool> used(nums.size(), false);
backtracking(nums, used);
return result;
}
};
if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
continue;
} //树层去重
if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == true) {
continue;
} //树枝去重