LeetCode 78 90 子集 子集II (回溯,遞迴)
阿新 • • 發佈:2018-10-31
1.子集
難度:中等
給定一組不含重複元素的整數陣列 nums,返回該陣列所有可能的子集(冪集)。
說明:解集不能包含重複的子集。
示例:
輸入: nums = [1,2,3]
輸出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
思路:該題利用遞迴即可解決,注意的是可將輸入先排下序。
class Solution { public: vector<vector<int>> subsets(vector<int>& nums) { sort(nums.begin(),nums.end()); vector<vector<int>> result; vector<int> items; backtrade(nums,result,items,0); return result; } void backtrade(vector<int> nums,vector<vector<int>> &result,vector<int> items,int start){ result.push_back(items); //注意這裡是把每次累積的item給push_back到陣列中 for(int i=start;i<nums.size();i++){ items.push_back(nums[i]); backtrade(nums,result,items,i+1); items.pop_back(); } } };
2. 子集II
難度:中等
給定一個可能包含重複元素的整數陣列 nums,返回該陣列所有可能的子集(冪集)。
說明:解集不能包含重複的子集。
示例:
輸入: [1,2,2]
輸出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
思路:其實這道題跟上面一道思路一樣,只是在遞迴呼叫的函式前加了一個判斷條件,若nums[i]==nums[i-1]的時候,直接跳過,避免重複。
class Solution { public: vector<vector<int>> subsetsWithDup(vector<int>& nums) { sort(nums.begin(),nums.end()); vector<vector<int>> result; vector<int> items; backtrade(nums,result,items,0); return result; } void backtrade(vector<int> nums,vector<vector<int>> &result,vector<int> items,int start){ result.push_back(items); for(int i=start;i<nums.size();i++){ if(i>start && nums[i]==nums[i-1]) continue; //加了判斷條件 items.push_back(nums[i]); backtrade(nums,result,items,i+1); items.pop_back(); } } };