【56.子集】

题目

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

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

示例 1：

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

题解

方式一：回溯

public List<List<Integer>> subsets(int[] nums) {
    List<List<Integer>> result = new LinkedList<>();
    backtrack(nums, 0, new LinkedList<>(), result);
    return result;
}

public void backtrack(int[] nums, int start, List<Integer> sub, List<List<Integer>> result) {
    result.add(new LinkedList<>(sub));
    for (int i = start; i < nums.length; i++) {
        // 选择
        sub.add(nums[i]);
        // 递归
        backtrack(nums, i + 1, sub, result);
        // 撤销
        sub.remove(sub.size() - 1);
    }
}

方式二：位运算

public List<List<Integer>> subsets(int[] nums) {
    int n = nums.length;
    List<List<Integer>> result = new LinkedList<>();
    // 子集个数：2^n
    for (int i = 0; i < (1 << n); i++) {
        List<Integer> sub = new LinkedList<>();
        // i的二进制位为1就包含到结果中
        for (int j = 0; j < n; j++) {
            if ((i & (1 << j)) != 0) {
                sub.add(nums[j]);
            }
        }
        result.add(sub);
    }
    return result;
}

总结

算法：回溯、位运算