今天继续学习回溯算法解决相关问题。
给你一个整数数组 nums ,找出并返回所有该数组中不同的递增子序列,递增子序列中 至少有两个元素 。你可以按 任意顺序 返回答案。
数组中可能含有重复元素,如出现两个整数相等,也可以视作递增序列的一种特殊情况。
491.递增子序列
示例 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]]
思路:
1.本题相比起Day27中的两道子集题目发生了变化,原本我们是将数组排序之后再进行树层去重以找到符合题意的子集;而在本题中我们不能对数组进行排序,否则数组的所有子集都是递增子序列了,因此只能在原数组中进行寻找。
2.本题中也涉及到去重的问题,无论是树枝还是树层都需要进行去重:(1)针对树枝去重,如果当前遍历的结点值小于目前子序列最后的元素的值,必须舍去;(2)针对树层去重,横向树层遍历时不能重复使用值相同的结点。
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);
//注意此处不能return,因为要取所有的结点
}
int used[201] = {0};//每一层单独创建一个used数组用来存储当前树层遍历时是否用了重复的数字
for(int i = startIndex; i < nums.size(); i++){
//进行树枝去重和树层去重,或之前为树枝去重
if((!path.empty() && nums[i] < path.back()) || used[nums[i] + 100] == 1){
continue;
}
path.push_back(nums[i]);
used[nums[i] + 100] = 1;
backtracking(nums, i + 1);
path.pop_back();
}
}
vector<vector<int>> findSubsequences(vector<int>& nums) {
backtracking(nums, 0);
return result;
}
};
启发:
1.因为本题中不能对原数组进行排序,因此不能用和Day27的两道子集一样通过一个used数组存储前一个变量是否被使用,然后通过nums[i] == nums[i - 1] && used[i - 1] = false去重,因为与当前元素相等的元素不一定相邻。作为替代我们为每一个树层创建一个used数组,用来存储这一层横向遍历时使用过的元素的值。
而关于树枝遍历我们只需要将当前元素的值与目前的子序列中最后的元素值进行比较即可。
(此处引用代码随想录中讲解的原图)
2.本题依旧属于求子集问题,因此需要取得的是所有的树形结构上的结点,而非只有叶子结点。
46.全排列
给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
示例 1:
输入:nums = [1,2,3]
输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
思路:
1.本题属于回溯算法中的全新题型了,在排列问题[1,2]和[2,1]是两种排列,在树层遍历时每一次都是从数组的起始位置开始遍历。因此我们需要一个布尔数组来判断当前位置上的元素在树枝递归时是否已经取过。
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;
}
path.push_back(nums[i]);
used[i] = true;
backtracking(nums,used);
used[i] = false;
path.pop_back();
}
}
vector<vector<int>> permute(vector<int>& nums) {
vector<bool> used(nums.size(), false);
backtracking(nums, used);
return result;
}
};
启发:
1.排列问题与集合内元素的排列顺序有关,因此每一层树层遍历都是从数组起始位置开始,而我们需要存储当前位置元素是否已经在排列集合中,因此需要一个布尔数组来记录。