1. 程式人生 > >LeetCode 491. 遞增子序列(C++)

LeetCode 491. 遞增子序列(C++)

給定一個整型陣列, 你的任務是找到所有該陣列的遞增子序列,遞增子序列的長度至少是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]。

給定陣列中可能包含重複數字,相等的數字應該被視為遞增的一種情況。

思路:採用迴圈遞迴的思想。

比如輸入[3, 4, 5, 6]

從3開始,放入tmp中,只要後面的數比tmp最後一個數大,則加入tmp

因此第一次結果分別為[3, 4]、[3, 4, 5]、 [3, 4, 5, 6]

當遞迴結束時,把tmp最後一個元素取出:

例如,當tmp=[3, 4, 5, 6]時,此時k=3,遞迴結束,將6取出,tmp=[3, 4, 5],

此時,k=2,i=k遞迴結束,取出5, tmp=[3, 4],接著i=3, tmp=[3, 4, 6]

因此最終輸出的順序是:[3, 4]、[3, 4, 5]、[3, 4, 5, 6]、[3, 4, 6]、 [3, 5]、[3, 5, 6]、 [3, 6]

[4,5]、[4,5,6]、[4,6]、[5,6]

C++

class Solution {
public:
    void DFS(vector<int>& nums, set<vector<int>>& res, vector<int>& tmp, int k)
    {
        if(tmp.size()>=2)
        {
            res.insert(tmp);
        }
        for(int i=k;i<nums.size();i++)
        {
            if(tmp.empty() || nums[i]>=tmp.back())
            {
                tmp.push_back(nums[i]);
                DFS(nums, res, tmp, i+1);
                tmp.pop_back();
            }
        }
    }
    vector<vector<int>> findSubsequences(vector<int>& nums) 
    {
        int n=nums.size();
        set<vector<int>> res;
        vector<int> tmp;
        DFS(nums,res,tmp,0);
        return vector<vector<int>>(res.begin(),res.end());       
    }
};