题目:491.非递减子序列
尝试解答:
在for循环中做判断,如果后一个数大于前一个数,那么就将其存进path,否则直接continue,也不同进入递归。在终止条件处判断,若path.size() >= 2,就收集结果。
而且同样要进行树层去重。
尝试代码如下:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex, vector<bool>& used){
if(path.size() >= 2){
result.push_back(path);
}
if(startIndex >= nums.size()) return;
for(int i = startIndex; i < nums.size(); i++){
if(i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) continue;
if(i > 0 && nums[i] < nums[i - 1]) continue;
path.push_back(nums[i]);
used[i] = true;
backtracking(nums, i + 1, used);
path.pop_back();
used[i] = false;
}
}
vector<vector<int>> findSubsequences(vector<int>& nums) {
vector<bool> used(nums.size(), false);
backtracking(nums, 0, used);
return result;
}
};
本以为秒了,但是好像被卡数据了?
其实没有意识到本题的难点是什么:
难点在于,本题不能对原数组进行排序,如果排序那么得到的非递增子序列和原数组是不一样的(因为要考虑增减性),但如果不进行排序,就无法用之前的方法对数组进行去重。
正确思路如下:
树型结构:
对于本题的去重,应该定义一个unordered<int> uset。
代码如下(对于哈希表还得加强理解、认识和运用):
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex){
if(path.size() >= 2){
result.push_back(path);
}
if(startIndex >= nums.size()) 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();
}
}
vector<vector<int>> findSubsequences(vector<int>& nums) {
backtracking(nums, 0);
return result;
}
};
这也是需要注意的点,unordered_set<int> uset;
是记录本层元素是否重复使用,新的一层uset都会重新定义(清空),所以要知道uset只负责本层!所以不需要对uset进行什么回溯操作。
优化:注意题目中说了,数值范围[-100,100],所以完全可以用数组来做哈希。
优化如下:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex){
if(path.size() >= 2){
result.push_back(path);
}
if(startIndex >= nums.size()) return;
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();
}
}
vector<vector<int>> findSubsequences(vector<int>& nums) {
backtracking(nums, 0);
return result;
}
};
同样,used只在本层起作用,下一层需要重新定义。而且used也不用进行回溯操作。