【LeetCode-491】递增子序列
问题
给定一个整型数组, 你的任务是找到所有该数组的递增子序列,递增子序列的长度至少是2。
示例
输入: [4, 6, 7, 7]
输出: [[4, 6], [4, 7], [4, 6, 7], [4, 6, 7, 7], [6, 7], [6, 7, 7], [7,7], [4,7,7]]
解答
class Solution {
public:
vector<vector<int>> findSubsequences(vector<int>& nums) {
dfs(nums, 0);
return res;
}
private:
vector<vector<int>> res;
vector<int> ans;
void dfs(vector<int>& nums, int pos) {
if (ans.size() > 1) res.push_back(ans);
for (int i = pos; i < nums.size(); i++) { // 保证输出无顺序要求
if (!ans.empty() && ans.back() > nums[i] || checkRepeat(nums, pos, i)) continue;
ans.push_back(nums[i]);
dfs(nums, i + 1);
ans.pop_back();
}
}
bool checkRepeat(vector<int>& nums, int pos, int i) { // 保证无序数组中当前递归深度不含重复数字
for (int j = pos; j < i; j++)
if (nums[i] == nums[j]) return true;
return false;
}
};
重点思路
题目要求返回所有的数组,第一时间想到回溯方法。分析本题的各种重复情况:
- 输入序列存在重复元素(输入
[1, 4, 7, 7]
中7
是重复的); - 输出序列内元素无顺序要求(本题输出
[4, 7]
与[7, 4]
是一样的)。
所以我们使用每层深度的循环开始位置pos
作为限制(保证输出无顺序要求),并保证当前层循环起始位置与当前位置(左闭右开)间是否存在与当前位置相同的元素(保证不统计输入序列的重复元素)。