子集
题目描述:
给你一个整数数组 nums
,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
思路:
一开始看见题目想的是枚举,但是感觉过于复杂。后来看了题目的相关标签是回溯,才意识到使用回溯来做很方便。
对于每一个元素,要么子集中包含它,要么不包含他,想象一颗二叉树,每个节点代表下标为深度值-1的元素,左子树值全为0(代表这个元素没有被取),右子树值全为1(代表该元素被选取),那么当数的深度到达nums.length时,就代表已经包含了所有元素被取与不被取的结果。
到达nums.length这一层(最后一层)的节点走过的路径就代表了每一个子集(比如最左下的叶子节点路径值全为0,代表所有元素都没有被取,即空集),所有路径组成的list就是我们最后要求的子集的集合。
利用树来看,我们的回溯函数return的条件为树的深度到达nums.length(在return前要把当前路径存储到res中),我们可以使用一个list来存储路径,k代表当前到达的深度,对于每一个节点,考虑取它和不取它的情况:
fun(tmpres,k+1,nums);//不取下标为k的元素
tmpres.add(nums[k]);
fun(tmpres,k+1,nums);//取下标为k的元素
需要注意在回退时要删去上一次添加进list的元素(即最新添加的那个元素):
tmpres.remove(tmpres.size()-1);
代码:
public static List<List<Integer>> res = new ArrayList<>();
public static List<List<Integer>> subsets(int[] nums) {
res.clear();
List<Integer> tmpres = new ArrayList<>();
fun(tmpres,0,nums);
return res;
}
public static void fun(List<Integer> tmpres,int k,int[] nums){
if(k==nums.length){
res.add(new ArrayList<>(tmpres));
return;
}
fun(tmpres,k+1,nums);
tmpres.add(nums[k]);
fun(tmpres,k+1,nums);
tmpres.remove(tmpres.size()-1);
}