LeetCode 78. 子集

题目描述

给定一组不含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。

说明：解集不能包含重复的子集。

示例:

输入: nums = [1,2,3]
输出:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

问题分析

此题和leetcode 77题很像，稍作改动即可。上一道题的数字对应着这道题数组的下标，所以存入out中的是nums[i]。然后上一道题out中的数的个数是固定的k个，这道题是从0到nums.size()都有，所以遍历一遍就可以了，逐个得到不同个数的组合即可。

代码实现

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> ans;
        vector<int> out;
        for(int k = 0; k <= nums.size(); k++)
            DFS(0, out, nums, k, ans);
        return ans;
    }
    
    void DFS(int left, vector<int>& out, vector<int>& nums, int& k, vector<vector<int>>& ans){
        if(out.size() == k){
            ans.push_back(out);
            return;
        }
        for(int i = left; i <= nums.size() - k + out.size(); i++){
            out.push_back(nums[i]);
            DFS(i + 1, out, nums, k, ans);
            out.pop_back();
        }
        return;
    }
};