代码随想录–回溯部分
day 26 回溯第四天
一、力扣491–递增子序列
代码随想录题目链接:代码随想录
给你一个整数数组 nums ,找出并返回所有该数组中不同的递增子序列,递增子序列中 至少有两个元素 。你可以按 任意顺序 返回答案。
数组中可能含有重复元素,如出现两个整数相等,也可以视作递增序列的一种特殊情况。
太复杂了,想直接用组合Ⅱ的方法做,但是一旦对nums排序了,那么任何子序列都是递增的,输出的结果就不对了,这个题的nums是不能有任何顺序操作的
那么去重要怎么做呢
那么就需要在每一层进行去重了
设置一个哈希表,记录本层用过的元素,下次遍历到的时候就不能再用了
相当于做了个nums[i]==nums[i-1]的判断
那么什么时候插入结果呢
这里和组合Ⅱ类似,应该每次递归的时候都加入自身,但是题目要求子序列有两个元素,所以要加个判断,当path长度大于1即至少为2时再加入
遍历时的剪枝操作有两个判断:path末尾和当前元素大小比较,要满足递增才继续;当前元素非重复
代码如下:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex)
{
if(path.size() > 1)
{
result.push_back(path);
}
int hash[201] = {0};
for(int i = startIndex; i < nums.size(); i ++)
{
if(!path.empty() && nums[i] < path.back() || hash[nums[i] + 100] == 1) continue;
hash[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;
}
};
二、力扣46–全排列
代码随想录题目链接:代码随想录
给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
用used数组记录搜过的数字就行,秒了
代码如下:
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
vector<bool> used;
void backTracking(vector<int> nums)
{
if(path.size() == nums.size())
{
result.push_back(path);
return;
}
for(int i = 0; i < nums.size(); i ++)
{
if(used[i])continue;
used[i] = true;
path.push_back(nums[i]);
backTracking(nums);
used[i] = false;
path.pop_back();
}
}
vector<vector<int>> permute(vector<int>& nums) {
used = vector<bool>(nums.size(), false);
backTracking(nums);
return result;
}
};
三、力扣47–全排列Ⅱ
代码随想录题目链接:代码随想录
给定一个可包含重复数字的序列 nums ,按任意顺序 返回所有不重复的全排列。
题意不太好理解,核心意思是nums里面能够包含重复的数字,那么这样的全排列就会有重复的结果,要对这个结果降重才行
那么思路上依然是需要全排列,然后在搜索的时候降重
步骤还是先排序,再遍历,但是不需要startIndex了,每层都从头遍历
注意全排列本身需要一个used数组,而降重也需要一个,是不是需要两个used数组呢
并不是,这两个数组是一样的,都是在递归前set成true,回溯后再回归false,都用于表示某个位置是否被使用了
所以只要在全排列的代码中加一行去重判断和一行排序即可
代码如下:
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
vector<bool> used;
void backTracking(vector<int> nums)
{
if(path.size() == nums.size())
{
result.push_back(path);
return;
}
for(int i = 0; i < nums.size(); i ++)
{
if(i > 0 && nums[i] == nums[i-1] && used[i-1] == false) continue; // Add one
if(used[i])continue;
used[i] = true;
path.push_back(nums[i]);
backTracking(nums);
used[i] = false;
path.pop_back();
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
sort(nums.begin(), nums.end()); // Add two
used = vector<bool>(nums.size(), false);
backTracking(nums);
return result;
}
};