题目来源:https://leetcode.cn/problems/permutations-ii/description/
C++题解1:这道题需要有两个不同的数组来标记使用过的元素,一个存放同层使用过的元素,根据元素数值来判断,范围为[-10,10],在同层更新即可;另一个数组存放纵向路径使用过的元素,根据元素位置来判断,范围为[0,7],需在下一层时传递更新数组,回溯时回到原来数组。
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int> nums, int* used_depth) {
if(path.size() == nums.size()) {
res.push_back(path);
return;
}
int used[21] = {0};
for(int i = 0; i < nums.size(); i++) {
if(used[nums[i] + 10] == 1 || used_depth[i] == 1) continue;
used[nums[i] + 10] = 1;
used_depth[i] = 1;
path.push_back(nums[i]);
backtracking(nums, used_depth);
path.pop_back();
used_depth[i] = 0;
}
return ;
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
int used_depth[8] = {0};
backtracking(nums, used_depth);
return res;
}
};
C++题解2(来源代码随想录):在对同层去重时,提前对nums进行排序,通过对比nums前后元素数值来实现去重。
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;
}
};