LeetCode 78 子集

题目描述

子集

给你一个整数数组 nums ，数组中的元素 互不相同 。返回该数组所有可能的

子集

（幂集）。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1：

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

示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

• 1 <= nums.length <= 10
• -10 <= nums[i] <= 10
• nums 中的所有元素 互不相同

解法

解法1：循环枚举

思想：依次循环每个元素，每次把前面的所有结果都加上当前元素即可

java代码：

class Solution {
    public List<List<Integer>> subsets(int[] nums) {
        ArrayList<List<Integer>> res = new ArrayList<>();
        res.add(new ArrayList<>());
        for (int num : nums) {
            int size = res.size();
            for (int i = 0; i < size; i++) {
                List<Integer> newSub = new ArrayList<>(res.get(i));
                newSub.add(num);
                res.add(newSub);
            }
        }
        return res;
    }
}

复杂度

• 时间复杂度：O(n×2^n)，一共 2^n个状态，每种状态需要 O(n) 的时间来构造子集。
• 空间复杂度：O(n)

解法2：递归枚举

class Solution {
    public List<List<Integer>> subsets(int[] nums) {
        ArrayList<List<Integer>> res = new ArrayList<>();
        ArrayList<Integer> element = new ArrayList<>();
        dfs(0, nums, element, res);
        return res;
    }

    private void dfs(int i, int[] nums, ArrayList<Integer> element, ArrayList<List<Integer>> res) {
        res.add(new ArrayList<>(element));
        for (int j = i; j < nums.length; j++) {
            element.add(nums[j]);
            dfs(j + 1, nums, element, res);
            // 每次循环后注意回退一个
            element.remove(element.size() - 1);
        }
    }
}