LeetCode78子集(dfs)

最新推荐文章于 2023-10-19 10:33:05 发布

我会jvav啊

最新推荐文章于 2023-10-19 10:33:05 发布

阅读量77

点赞数

分类专栏： leetcode 文章标签： dfs leetcode java 算法回溯

本文链接：https://blog.csdn.net/qq_43434328/article/details/112972218

版权

leetcode 专栏收录该内容

84 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何使用深度优先搜索（DFS）解决LeetCode第78题——子集问题。作者通过分享初始的繁琐代码和后续的优化过程，阐述了如何简化vis数组和步数记录，以提高代码效率。

摘要由CSDN通过智能技术生成

在这里插入图片描述
题面一看就是dfs 每一步dfs都把路径存下来即可
第一遍代码较为繁琐后面有优化

class Solution {
    List<List<Integer>> lists = new ArrayList<>();
    public List<List<Integer>> subsets(int[] nums) {
        boolean[] vis = new boolean[nums.length];
        dfs(new ArrayList<>(),nums,vis,0,0);
        return lists;
    }
    public void dfs(List<Integer> list,int[] nums,boolean[] vis,int step,int fori){
        List<Integer> temp = new ArrayList<>(list);
        lists.add(temp);
        if (step == nums.length) return;
        for (int i = fori; i < nums.length; i++) {
            if (vis[i]) continue;
            list.add(nums[i]);
            vis[i] = true;
            dfs(list,nums,vis,step+1,i+1);
            list.remove(step);
            vis[i] = false;
        }
    }
}

第一次遍写完发现有许多用不到的地方我们进行代码量的优化。
由于每次只继续拼接身后所以vis数组可以省去
也不用记录当前步数也可以省去 step
优化后

class Solution {
    List<List<Integer>> lists = new ArrayList<>();
    public List<List<Integer>> subsets(int[] nums) {
        dfs(new ArrayList<>(),nums,0);
        return lists;
    }
    public void dfs(List<Integer> list,int[] nums,int fori){
        List<Integer> temp = new ArrayList<>(list);
        lists.add(temp);
        for (int i = fori; i < nums.length; i++) {
            list.add(nums[i]);
            dfs(list,nums,i+1);
            list.remove(list.size()-1);
        }
    }
}