1. 程式人生 > >LeetCode 78 90 子集 子集II (回溯,遞迴)

LeetCode 78 90 子集 子集II (回溯,遞迴)

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();
        }
    }
};