问题描述

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

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

示例:

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

输出:

[

[3],

[1],

[2],

[1,2,3],

[1,3],

[2,3],

[1,2],

[]

]

回溯解决解决

在前面一道题450,什么叫回溯算法,一看就会,一写就废中提到过子集的问题,这里再来看一下,回溯的模板如下,就是先选择,最后再撤销

private void backtrack("原始参数") { //终止条件(递归必须要有终止条件) if ("终止条件") { //一些逻辑操作(可有可无,视情况而定) return; } for (int i = "for循环开始的参数"; i < "for循环结束的参数"; i ) { //一些逻辑操作(可有可无,视情况而定) //做出选择 //递归 backtrack("新的参数"); //一些逻辑操作(可有可无,视情况而定) //撤销选择 } }

这道题也一样,可以把它想象成为一颗n叉树,通过DFS遍历这棵n叉树,他所走过的所有路径都是子集的一部分,看下代码

public List<List<Integer>> subsets(int[] nums) { List<List<Integer>> list = new ArrayList<>(); backtrack(list, new ArrayList<>(), nums, 0); return list; } private void backtrack(List<List<Integer>> list, List<Integer> tempList, int[] nums, int start) { //走过的所有路径都是子集的一部分,所以都要加入到集合中 list.add(new ArrayList<>(tempList)); for (int i = start; i < nums.length; i ) { //做出选择 tempList.add(nums[i]); //递归 backtrack(list, tempList, nums, i 1); //撤销选择 tempList.remove(tempList.size() - 1); } }

因为在第450题刚讲过这道题,所以基本上没什么难度,其实这道题还可以使用位运算解决,来看下

位运算解决

数组中的每一个数字都有选和不选两种状态,我们可以用0和1表示,0表示不选,1表示选择。如果数组的长度是n,那么子集的数量就是2^n。比如数组长度是3,就有8种可能,分别是

[0,0,0]

[0,0,1]

[0,1,0]

[0,11]

[1,0,0]

[1,0,1]

[11,0]

[111]

这里参照示例画个图来看下

去空集的子集个数(回溯和位运算解子集)(1)

public static List<List<Integer>> subsets(int[] nums) { //子集的长度是2的nums.length次方,这里通过移位计算 int length = 1 << nums.length; List<List<Integer>> res = new ArrayList<>(length); //遍历从0到length中间的所有数字,根据数字中1的位置来找子集 for (int i = 0; i < length; i ) { List<Integer> list = new ArrayList<>(); for (int j = 0; j < nums.length; j ) { //如果数字i的某一个位置是1,就把数组中对 //应的数字添加到集合 if (((i >> j) & 1) == 1) list.add(nums[j]); } res.add(list); } return res; }

非递归解决

这题还有其他解题思路,比如先加入一个空集让他成为新的子集,然后每遍历一个元素就在原来的子集的后面追加这个值。还以示例来分析下

去空集的子集个数(回溯和位运算解子集)(2)

public List<List<Integer>> subsets(int[] nums) { List<List<Integer>> res = new ArrayList<>(1 << nums.length); //先添加一个空的集合 res.add(new ArrayList<>()); for (int num : nums) { //每遍历一个元素就在之前子集中的每个集合追加这个元素,让他变成新的子集 for (int i = 0, j = res.size(); i < j; i ) { //遍历之前的子集,重新封装成一个新的子集 List<Integer> list = new ArrayList<>(res.get(i)); //然后在新的子集后面追加这个元素 list.add(num); //把这个新的子集添加到集合中 res.add(list); } } return res; }

如果非要把它改为递归的也是可以的,仅仅提供了一种思路,有兴趣的也可以看下

public List<List<Integer>> subsets(int[] nums) { List<List<Integer>> res = new ArrayList<>(1 << nums.length); res.add(new ArrayList<>()); recursion(nums, 0, res); return res; } public static void recursion(int[] nums, int index, List<List<Integer>> res) { //数组中的元素都访问完了,直接return if (index >= nums.length) return; int size = res.size(); for (int j = 0; j < size; j ) { List<Integer> list = new ArrayList<>(res.get(j)); //然后在新的子集后面追加一个值 list.add(nums[index]); res.add(list); } //递归下一个元素 recursion(nums, index 1, res); }

其他解决方式

在426,什么是递归,通过这篇文章,让你彻底搞懂递归中最后讲到分支污染的时候提到过这样一个问题:生成一个2n长的数组,数组的值从0到(2n)-1。我们可以把它想象成为一颗二叉树,每个节点的子树都是一个可选一个不可选

去空集的子集个数(回溯和位运算解子集)(3)

所以我们也可以参照这种方式来写,代码如下

public List<List<Integer>> subsets(int[] nums) { List<List<Integer>> res = new ArrayList<>(); helper(res, nums, new ArrayList<>(), 0); return res; } private void helper(List<List<Integer>> res, int[] nums, List<Integer> list, int index) { //终止条件判断 if (index == nums.length) { res.add(new ArrayList<>(list)); return; } //每一个节点都有两个分支,一个选一个不选 //走不选这个分支 helper(res, nums, list, index 1); //走选择这个分支 list.add(nums[index]); helper(res, nums, list, index 1); //撤销选择 list.remove(list.size() - 1); }

总结

这题难度不大,但解法比较多,上面介绍的每一种基本上都是一种新的解题思路,如果能全部掌握将会有很大收获。

去空集的子集个数(回溯和位运算解子集)(4)

,