代码随想录算法训练营第29天|491.递增子序列、 46.全排列、47.全排列 II
491.递增子序列
子序列--> 子集,然后就想到会不会是回溯
使用set进行去重
// 使用set进行去重
class Solution {
private:
vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
// 因为需要遍历完,所以不需要中止条件
if (path.size() > 1) {
res.push_back(path);
}
unordered_set<int> uset;
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) {
res.clear();
path.clear();
backtracking(nums, 0);
return res;
}
};
考虑到题干给出的 nums[i] 的范围是 [-100, 100], 那么就可以考虑直接使用数组来模拟哈希的过程,相比于使用set的效率更高。
// 使用数组进行去重
class Solution {
private:
vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
// 因为需要遍历完,所以不需要中止条件
if (path.size() > 1) {
res.push_back(path);
}
int used[201] = {0};
for (int i = startIndex; i < nums.size(); ++i) {
if ((!path.empty() && nums[i] < path.back()) || used[nums[i] + 100] == 1) continue;
used[nums[i] + 100] = 1;
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> findSubsequences(vector<int>& nums) {
res.clear();
path.clear();
backtracking(nums, 0);
return res;
}
};
46.全排列
相比于组合问题,排列问题的顺序也很重要。准备使用used数组来记录元素是否被使用过。
class Solution {
private:
vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int>& nums, vector<bool> used) {
// 终止条件
if (path.size() == nums.size()) {
res.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);
used[i] = false;
path.pop_back();
}
}
public:
vector<vector<int>> permute(vector<int>& nums) {
res.clear();
path.clear();
vector<bool> used(nums.size(), false);
backtracking(nums, used);
return res;
}
};
47.全排列 II
这题里面使用 nums[i - 1] == true 或者 false,都是可以的,只是去重的过程稍微有不一样。 nums[i - 1] == false, 说明是在同一层使用过,去重效率更高
class Solution {
private:
vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int>& nums, vector<bool>& used) {
// 终止条件
if (path.size() == nums.size()) {
res.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);
used[i] = false;
path.pop_back();
}
}
}
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
res.clear();
path.clear();
sort(nums.begin(), nums.end());
vector<bool> used(nums.size(), false);
backtracking(nums, used);
return res;
}
};