LeetCode 491. 遞增子序列(C++)
阿新 • • 發佈:2018-12-16
給定一個整型陣列, 你的任務是找到所有該陣列的遞增子序列,遞增子序列的長度至少是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()); } };