第七章回溯算法part05
491. 递增子序列
给你一个整数数组 nums
,找出并返回所有该数组中不同的递增子序列,递增子序列中 至少有两个元素 。你可以按 任意顺序 返回答案。
数组中可能含有重复元素,如出现两个整数相等,也可以视作递增序列的一种特殊情况。
示例 1:
输入:nums = [4,6,7,7] 输出:[[4,6],[4,6,7],[4,6,7,7],[4,7],[4,7,7],[6,7],[6,7,7],[7,7]]
示例 2:
输入:nums = [4,4,3,2,1] 输出:[[4,4]]
思路:一开始被两个示例测试用例带偏,只考虑到增加过程中插入元素后者<前者且忽视了相同元素不连续的情况,所以其他测试用例结果错误。
错误代码如下:
for(int i=startIndex;i<nums.size();i++){
if(i>startIndex&&nums[i]<=nums[i-1]){
continue;
}
if(i>0&&i==startIndex&&nums[i]<nums[i-1]){
continue;
}
path.push_back(nums[i]);
backtracking(nums,i+1);
path.pop_back();
}
如果测试例子是nums=[4,7,6,7]则不能通过,因此发现问题在于应该确保元素在书的同一层前面所有使用过的元素中都不能重复。于是使用vector带的find函数在path中查找,结果还是报错。之前做的时候也是感觉这种思路可以但是答案出错。???????不知道为什么
for(int i=startIndex;i<nums.size();i++){
if(i>startIndex&&find(path.begin(),path.end(),nums[i])!=path.end()){
continue;
}
if(i>0&&i==startIndex&&nums[i]<path.back()){
continue;
}
path.push_back(nums[i]);
backtracking(nums,i+1);
path.pop_back();
}
题解使用了一个unordered_set存储每一层使用过的元素,使用find查找,插入元素判断条件是path是否为空。
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;
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可以改为哈希表存储,查找复杂度有O(n)降为O(1).
int used[201]={0};
for(int i=startIndex;i<nums.size();i++){
if(i>startIndex&&used[nums[i]+100]!=0){
continue;
}
if(!path.empty()&&nums[i]<path.back()){
continue;
}
used[nums[i]+100]++;
//uset.insert(nums[i]);
path.push_back(nums[i]);
backtracking(nums,i+1);
path.pop_back();
}
46. 全排列
给定一个不含重复数字的数组 nums
,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
思路:一开始用了前面的做法创建一个哈希表存储同一层用过的元素,大错特错!!!!应该是树的同一分支不能有相同元素,应该把用过的元素放到回溯函数中!!!!!
要区分好到底是同一层还是同一分支元素不能重复啊!
class Solution {
public:
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++){
//同一分支的元素不能重复,因此要放到回溯函数的参数中并实现回溯
if(used[i]==true){
continue;
}
used[i]=true;
path.push_back(nums[i]);
backtracking(nums,used);
path.pop_back();
used[i]=false;
}
}
vector<vector<int>> permute(vector<int>& nums) {
vector<bool> used(nums.size(),false);
backtracking(nums,used);
return result;
}
};
47. 全排列 II
给定一个可包含重复数字的序列 nums
,按任意顺序 返回所有不重复的全排列。
示例 1:
输入:nums = [1,1,2] 输出: [[1,1,2], [1,2,1], [2,1,1]]
前两题的综合版,重点还是理解两个关键问题:
①同一层元素不能重复, 一般是解决问题给出的序列中有重复元素的问题;使用哈希表做记录,不作为回溯参数。
int uset[21]={0};
②同一分支的元素不能重复,解决问题给出的序列中每个元素不能重复使用,只能使用一次的问题,使用了一个数组做记录,且必须被用作回溯函数的参数以实现回溯。
void backtracking(vector<int>&nums,vector<bool> used){
理解了上面两个关键问题,这道题就一遍ac了
class Solution {
public:
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;
}
int uset[21]={0}; //记录同一层使用过的元素
for(int i=0;i<nums.size();i++){
if(used[i]==true||uset[nums[i]+10]!=0) continue;
uset[nums[i]+10]++;
used[i]=true; //记录同一分支使用过的元素并实现回溯
path.push_back(nums[i]);
backtracking(nums,used);
path.pop_back();
used[i]=false;
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
vector<bool> used(nums.size(),false);
backtracking(nums,used);
return result;
}
};