给你一个整数数组 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]]
思路:本题使用回溯算法,分析题目可知,本题需要找递增子序列,所以不能对数组进行排序。
另外,数组中含有重复元素,所以不做去重的结果集里面可能含有重复序列,所以要在每一层for循环横向遍历时,对这一层的元素去重。
定义全局变量result为结果集,path为收集到的递增序列。for循环横向遍历,递归纵向遍历,把它想象成树状结构,收集每一个节点值。
回溯三部曲:
- 确定回溯函数参数和返回值:回溯函数的参数有集合nums,每一层for循环起始搜索位置startIndex。回溯函数的返回类型一般都会void类型。
- 确定回溯函数的终止条件:如果startIndex大于等于nums.size(),说明搜索到了叶节点,return。这一条也可以不用写,因为for循环的最大边界就是nums.size()。
- 确定单层搜索的过程:首先定义变量uset,for循环横向遍历,递归过程纵向遍历,for循环每次从startIndex开始遍历。如果当前元素值比path的最后一个值小,或者当前元素在uset中出现过,跳过,继续下一个元素。否则,将当前元素值加入path,递归函数不断调用自己往深处遍历。然后递归函数下面要写回溯部分,撤掉本次处理的结果,path弹出最后一个值。
代码:
class Solution { //491. 递增子序列
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int> const& nums, int startIndex) {
if (path.size() > 1) {
result.push_back(path);
}
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;
}
};
参考资料:代码随想录