LeetCode: 子集_子集leetcode-CSDN博客

本文链接：https://blog.csdn.net/mdzzname/article/details/100046753

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

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

示例:

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

class Solution {
public:
    // n 个元素的子集个数共 2^n个， 可用 n位二进制表示，每位表示该数是否在集合里面，遍历所有的数并加入到集合中
    vector<vector<int>> subsets(vector<int>& nums) {
        int m = nums.size();
        vector<vector<int>> ans;
        vector<int> temp;
        for(int i = 0; i < (1 << m); i++){
            for(int j = 0; j < m; j++)
                if((i >> j)&1)
                    temp.push_back(nums[j]);
            ans.push_back(temp);
            temp.clear();
        }
        return ans;
    }
};

class Solution {
public:
    // 其实子集结果形成了一棵数，每个index根据要不要加入子集集合形成两个分支 dfs到达叶子节点就一路上的子集加入到结果集中
    vector<vector<int>> subsets(vector<int>& nums) {
        int m = nums.size();
        vector<vector<int>> ans;
        vector<int> temp;
        dfs(nums, ans, temp, 0, m);
        return ans;
    }
    
    void dfs(vector<int>& nums, vector<vector<int>>& ans, vector<int>& temp, int index, int m){
        if(index>=m){
            ans.push_back(temp);
            return;
        }
        temp.push_back(nums[index]);
        dfs(nums, ans, temp, index+1, m);
        temp.pop_back();
        dfs(nums, ans, temp, index+1, m);
    }
};