491.递增子序列
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
if (path.size() > 1) {
result.push_back(path);
// 注意这里不要加return,要取树上的节点
}
unordered_set<int> uset; // 使用set对本层元素进行去重
for (int i = startIndex; i < nums.size(); i++) {
if ((!path.empty() && nums[i] < path.back())
|| uset.find(nums[i]) != uset.end()) {
continue;
}
uset.insert(nums[i]); // 记录这个元素在本层用过了,本层后面不能再用了
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> findSubsequences(vector<int>& nums) {
result.clear();
path.clear();
backtracking(nums, 0);
return result;
}
};
这道题也是需要用到去重技术,但由于这道题是要求递增子序列,因此不可以使用之前的先排序再使用数组来进行重复判断,而需要使用set来进行去重。
首先是终止条件,在路径大于等于2时就加入结果中即可。在遍历过程中,如果每一层中有重复的元素或者小于路径中最后一个值的元素就跳过该次。如果满足条件的话就在路径中加入该元素,同时也要在set中加入该元素。之后就是调用递归函数。在递归返回之后,回溯的过程只需要将path的元素弹出,而set中的元素由于是在每次循环中才创建的,记录每一层使用过的元素,而不是全局变量,因此不需要将其中的元素弹出!
46.全排列
class Solution {
private:
vector<int> path;
vector<vector<int>> result;
void backTracking(const vector<int>& nums, vector<bool>& used) {
if (path.size() == nums.size()) {
result.push_back(path);
return;
}
for (int i = 0; i < nums.size(); i++) {
if (used[i] == true) continue;
used[i] = true;
path.push_back(nums[i]);
backTracking(nums, used);
path.pop_back();
used[i] = false;
}
}
public:
vector<vector<int>> permute(vector<int>& nums) {
vector<bool> used(nums.size(), false);
backTracking(nums, used);
return result;
}
};
求全排列代码,不需要去重,也不需要startIndex来记录开始下标,因为每次循环都要从0开始,这样才能得到完成的排列。
使用used数组来记录在每次循环中使用过的元素,每次循环(每一层)都是从0到size-1,会遍历整个数组,在每个枝干中寻找未添加的元素,添加完元素后会将used数组更新,回溯后要重置path及used。
47.全排列 II
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)来判断是否重复即可。