当前位置:   article > 正文

「递归算法」:子集(两种解法)

「递归算法」:子集(两种解法)

一、题目

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

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

示例 1:

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

示例 2:

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

二、思路解析

解法一

来看看这道题的决策树模型,下图的 × 代表不选择该元素,✓ 代表选择,所有叶子节点均为答案。

而这种方法,需要一个 pos 变量,用于记录已经遍历过的元素个数。

当遍历个数和 nums 数组的大小一致时,则为函数出口,把一次遍历也插入到 ret 变量中,然后直接返回即可。

函数体部分代码也较为简单,剪枝我也在前面出过一期博客,在此不做讲解。

第一种解法的所有叶子结点均为符合题意的答案。

解法二

和解法一的区别,主要在于解法二用了一个 for 循环来遍历数组,且在循环中再次递归调用。

但是在这种解法,我们不再判断选与不选,而是把当前元素及其后面的每个元素都统计在内。

这样的结果,就是所有节点均为满足题意的答案,也就是图中绿色部分元素。

三、完整代码

解法一:

  1. class Solution {
  2. List<List<Integer>> ret;
  3. List<Integer> path;
  4. public List<List<Integer>> subsets(int[] nums) {
  5. ret = new ArrayList<>();
  6. path = new ArrayList<>();
  7. dfs(nums , 0);
  8. return ret;
  9. }
  10. public void dfs(int[] nums , int pos){
  11. // 解法一
  12. if(pos == nums.length){
  13. ret.add(new ArrayList<>(path));
  14. return;
  15. }
  16. //
  17. path.add(nums[pos]);
  18. dfs(nums , pos + 1);
  19. path.remove(path.size() - 1);
  20. // 不选
  21. dfs(nums , pos + 1);
  22. }

解法二:

  1. class Solution {
  2. List<List<Integer>> ret;
  3. List<Integer> path;
  4. public List<List<Integer>> subsets(int[] nums) {
  5. ret = new ArrayList<>();
  6. path = new ArrayList<>();
  7. dfs(nums , 0);
  8. return ret;
  9. }
  10. public void dfs(int[] nums , int pos){
  11. // 解法二
  12. ret.add(new ArrayList<>(path));
  13. for(int i = pos ; i < nums.length ; i ++){
  14. path.add(nums[i]);
  15. dfs(nums , i + 1);
  16. path.remove(path.size() - 1);
  17. }
  18. }

以上就是本篇博客的全部内容啦,如有不足之处,还请各位指出,期待能和各位一起进步!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/2023面试高手/article/detail/102204
推荐阅读
相关标签
  

闽ICP备14008679号