leetcode491. 递增子序列/dfs

题目: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]]

说明:

  • 给定数组的长度不会超过15。
  • 数组中的整数范围是 [-100,100]。
  • 给定数组中可能包含重复数字,相等的数字应该被视为递增的一种情况。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/increasing-subsequences
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

基本思想:dfs

这里要说明一下:题目要求的是递增子序列,子序列和子串的唯一区别是:子序列可以不连续。这里很明显:子序列要保持原来数组中每个数字的相对顺序。

借助set去重:

class Solution {
    
    
public:
    set<vector<int>> res;
    vector<vector<int>> findSubsequences(vector<int>& nums) {
    
    
        vector<vector<int>> vres;
        
        if(nums.size() < 2)
            return vres;
        //子序列要保持相对顺序
        vector<int> cur;
        dfs(nums, 0, cur);
        
        for(auto r : res){
    
    
            vres.push_back(r);
        }
        return vres;
    }
    void dfs(vector<int>& nums, int pos, vector<int> cur){
    
    
        if(cur.size() > 1)
            res.insert(cur);
        for(int i = pos; i < nums.size(); ++i){
    
    
            if(cur.size() == 0 || nums[i] >= cur.back()){
    
    
                cur.push_back(nums[i]);
                dfs(nums, i + 1, cur);
                cur.pop_back();
            }
        }
    }
};

在dfs的过程中进行剪枝去重:
通过回溯时,保存弹出的元素实现,保证弹出的元素不会再进入当前数组中。

class Solution {
    
    
public:
    vector<vector<int>> res;
    vector<vector<int>> findSubsequences(vector<int>& nums) {
    
    
        if(nums.size() < 2)
            return res;
        vector<int> cur;
        dfs(nums, 0, cur);
        return res;
    }
    void dfs(vector<int>& nums, int pos, vector<int> cur){
    
    
        if(cur.size() > 1)
            res.push_back(cur);
        unordered_set<int> help;
        for(int i = pos; i < nums.size(); ++i){
    
    
            
            if((cur.size() == 0 || nums[i] >= cur.back()) && help.find(nums[i]) == help.end()){
    
    
                cur.push_back(nums[i]);
                dfs(nums, i + 1, cur);
                cur.pop_back();
                //去掉重复的元素
                help.insert(nums[i]);
            }
            
        }
    }
};

猜你喜欢

转载自blog.csdn.net/qq_31672701/article/details/108220207