【算法题解】78. 子集 - Java - 暴力、回溯、DFS

题目描述

给你一个整数数组 nums ，返回该数组所有可能的子集（幂集）。解集不能包含重复的子集。

 
示例 1：

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

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

Java代码

1. 暴力

    // 暴力
    public List<List<Integer>> subsets(int[] nums) {
        if (nums.length == 0) return null;
        
        List<List<Integer>> res = new ArrayList<>();
        res.add(new ArrayList<>());// 加入空集
        for (int num: nums) {
            int len = res.size();
            for (int i = 0; i < len; i++) {
                List<Integer> cur = new ArrayList<>(res.get(i));
                cur.add(num);
                res.add(cur);
            }
        }
        return res;
    }

2. 回溯

    // 回溯
    public List<List<Integer>> subsets(int[] nums) {
        if (nums.length == 0||nums == null) return null;
        // 返回的结果
        List<List<Integer>> res = new ArrayList<>();
        // 子集长度每次都遍历，0 -> length , 其中len 就是每一次遍历的长度，此处遍历目的是为了让len每次加一，也可以不用此方式
        for (int len = 0; len <= nums.length; len++) 
            backtracking(nums, len ,0, res, new ArrayList<Integer>());
        return res;
    }

    // 回溯。参数： startIndex从哪个下标开始
    public void backtracking(int[] nums, int len, int startIndex, List<List<Integer>> res, List<Integer> cur) {
        // 递归终止条件，每次递归的子集的长度长0 - length; 
        if (cur.size() == len) {
            res.add(new ArrayList<>(cur));
            return;
        }

        // 从上一层的startIndex 开始
        for (int i = startIndex; i < nums.length; i++) {
            // 排除不合法
            if (cur.contains(nums[i])) continue;
            // 做选择，加入
            cur.add(nums[i]);
            // 进入下一层
            backtracking(nums, len , i + 1, res, cur);
            // 取消选择，
            cur.remove(cur.size() - 1);
        }
    }

3. DFS

    // DFS
    public List<List<Integer>> subsets(int[] nums) {
         if (nums.length == 0||nums == null) return null;
        // 返回的结果
        List<List<Integer>> res = new ArrayList<>();
        dfs(nums, 0, res, new ArrayList<>());
        return res;
    }

    //DFS
    public void dfs(int[] nums, int startIndex, List<List<Integer>> res, List<Integer> cur) {
        res.add(new ArrayList<>(cur));

        for (int i = startIndex; i < nums.length; i++) {
            cur.add(nums[i]);
            dfs(nums, i + 1, res, cur);
            cur.remove(cur.size() - 1);
        }
    }