DFS回溯 选与不选
我们遍历每个元素,都可以考虑要不要将其放入到一个子集中
最直观的,数组[1,2,3]
我们将每个元素都放入子集,那么就得出了答案之一[1,2,3]
我们啥都不选,那么就得出了另外一个答案[ ];
我们只选1和2,得到答案[1,2]
这样看来,选与不选,我们将这两种情况可以列成一颗二叉树来表示
细看这颗二叉树的叶子结点,就是答案所在
那么我们只要构造出这颗二叉树然后遍历出叶子结点的结果就OK了
class Solution {
List<Integer> tmp = new ArrayList<Integer>();//存放子集的答案之一
List<List<Integer>> res = new ArrayList<>();//存放所有子集,也是我们返回的结果集
public List<List<Integer>> subsets(int[] nums) {
dfs(nums,0);
return res;
}
public void dfs(int[] nums,int n){//n用来控制层数,也就是我们需要的叶子结点层
if(n == nums.length){
res.add(new ArrayList<>(tmp));//这里我们需要复制一个tmp数组加入到结果集中,因为JAVA是值传递,我们传递了tmp的地址值,因此后续的操作都会影响到我们已经加入到结果集中的tmp,我们不能直接写res.add(tmp);
return;
}
tmp.add(nums[n]);
dfs(nums,n+1);
tmp.remove(tmp.size() - 1);
/*这里的删除操作就是回溯,比如我们将1,2,3加入到了tmp中,叶子结点的遍历并没有完成,我们需要回到上一个状态就是[1,2]的状态,然后不选择[3],达成[1,2]这个结果加入到res结果集中,然后再回到上个状态...
*/
dfs(nums,n+1);
}
}
还有类似的视情况构建二叉树然后遍历的题目:
李白打酒