leetcode-78-子集

78. 子集

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

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

示例 1：

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

示例 2：

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

方法一：迭代法实现子集枚举

思路与算法

因为在这数组中没有重复的元素，所有我们可以利用二进制的0,1表示元素是否出现，从而进行组合。

记原序列中元素的总数为 n。原序列中的每个数字 ai 的状态可能有两种，即「在子集中」和「不在子集中」。我们用 1 表示「在子集中」，0 表示不在子集中，那么每一个子集可以对应一个长度为 n 的 0/1 序列，第 i 位表示 ai 是否在子集中。例如，n=3 ，a={5,2,9} 时：

可以发现 0/1 序列对应的二进制数正好从 000 到 2^n - 1。我们可以枚举 mask∈[0,2n−1]，mask 的二进制表示是一个 0/1 序列，我们可以按照这个 0/1 序列在原集合当中取数。当我们枚举完所有 2^n 个 mask，我们也就能构造出所有的子集。

代码实现

class Solution {
    List<Integer> t = new ArrayList<Integer>();
    List<List<Integer>> ans = new ArrayList<List<Integer>>();

    public List<List<Integer>> subsets(int[] nums) {
        int n = nums.length;
        //1 << n  1左移n位确定循环的次数
        for (int mask = 0; mask < (1 << n); ++mask) {
            //每次刷新t
            t.clear();
            for (int i = 0; i < n; ++i) {
                //比较mask的二进制的每一位是否为1，是1就加入相应位置的数
                if ((mask & (1 << i)) != 0) {
                    t.add(nums[i]);
                }
            }
            //将t加入集合
            ans.add(new ArrayList<Integer>(t));
        }
        return ans;
    }
}

方法二：递归法实现子集枚举

思路与算法

我们也可以用递归来实现子集枚举。

假设我们需要找到一个长度为 n 的序列 a 的所有子序列，代码框架是这样的：

List<Integer> t = new ArrayList<Integer>();
public void dfs(int cur, int[] nums) {
        if (cur == nums.length) {
            // 记录答案
        	// ...
        }
    	// 考虑选择当前位置
        t.add(nums[cur]);
        dfs(cur + 1, nums);
    	// 考虑不选择当前位置
        t.remove(t.size() - 1);
        dfs(cur + 1, nums);
    }

对于 cur 位置，我们需要考虑 a[cur] 取或者不取，如果取，我们需要把 a[cur]放入一个临时的答案数组中（即上面代码中的 t，再执行 dfs(cur+1,n)，执行结束后需要对 t 进行回溯；如果不取，则直接执行 dfs(cur+1,n)。当 cur 增加到 n 的时候，记录答案并终止递归。

class Solution {
    List<Integer> t = new ArrayList<Integer>();
    List<List<Integer>> ans = new ArrayList<List<Integer>>();

    public List<List<Integer>> subsets(int[] nums) {
        //开始递归，从位置0开始
        dfs(0, nums);
        return ans;
    }

    public void dfs(int cur, int[] nums) {
        if (cur == nums.length) {
            // 记录答案
            ans.add(new ArrayList<Integer>(t));
            return;
        }
        // 考虑选择当前位置
        t.add(nums[cur]);
        dfs(cur + 1, nums);
        // 不考虑选择当前位置
        t.remove(t.size() - 1);
        dfs(cur + 1, nums);
    }
}