力扣刷题78. 子集

78. 子集 - 力扣(LeetCode)

使用dfs,用一个变量记录当前开始遍历的位置,下一次递归从当前位置+1开始递归,然后加入集合中,可以保证不会重复,回溯时直接移除最后的元素即可。

最开始的代码是没有记录位置的变量,每一次循环递归判断当前集合是否在目的集合中,而且需要排序,导致时间复杂度过高,而且在dfs中进行排序后,在移除最后的数据,数据会发生变化,导致移除了不该移除的数据导致结果错误。如果不进行排序,而在主函数中在遍历所有的结果进行排序移除重复集合,时间复杂度会过高导致超时。

超时的代码:

java 复制代码
class Solution {
    List<List<Integer>> list;
    List<Integer> res;

    public List<List<Integer>> subsets(int[] nums) {
         res = new LinkedList<>();
        list = new LinkedList<>();
        list.add(new LinkedList<>());
        for (int i = 0; i < nums.length; i++) {
            res = new LinkedList<>();
            res.add(nums[i]);
            dfs(res,nums);
        }
        List<List<Integer>> result = new LinkedList<>();
        for (int i = 0; i < list.size(); i++) {
            List<Integer> list1 = list.get(i);
            Collections.sort(list1);
            if (!result.contains(list1)) {
                result.add(list1);
            }
        }
        return result;
    }
    public void dfs(List<Integer> res,int[] nums) {
        if (!list.contains(res) && res.size() <= nums.length) {

            list.add(new LinkedList<>(res));
            System.out.println(res);
        }
        for (int i = 0; i < nums.length; i++) {
            if (!res.contains(nums[i])) {
                res.add(nums[i]);
                dfs(res,nums);
                res.remove(res.size() - 1);
            }
        }
    }
}

优化后的代码:

java 复制代码
class Solution {
        static List<List<Integer>> list;
    static List<Integer> res;
    public List<List<Integer>> subsets(int[] nums) {
        res = new LinkedList<>();
        list = new LinkedList<>();
        // int[] nums = new int[]{1, 2, 3};
        int index = 0;
        dfs(res, nums, index);
        return list;
    }
public static void dfs(List<Integer> res, int[] nums, int index) {
        list.add(new LinkedList<>(res));

        for (int i = index; i < nums.length; i++) {
            res.add(nums[i]);
            dfs(res, nums, i + 1);
            res.remove(res.size()-1);
        }
    }
}
相关推荐
那个村的李富贵6 小时前
CANN加速下的AIGC“即时翻译”:AI语音克隆与实时变声实战
人工智能·算法·aigc·cann
power 雀儿7 小时前
Scaled Dot-Product Attention 分数计算 C++
算法
琹箐7 小时前
最大堆和最小堆 实现思路
java·开发语言·算法
renhongxia18 小时前
如何基于知识图谱进行故障原因、事故原因推理,需要用到哪些算法
人工智能·深度学习·算法·机器学习·自然语言处理·transformer·知识图谱
坚持就完事了8 小时前
数据结构之树(Java实现)
java·算法
算法备案代理8 小时前
大模型备案与算法备案,企业该如何选择?
人工智能·算法·大模型·算法备案
赛姐在努力.8 小时前
【拓扑排序】-- 算法原理讲解,及实现拓扑排序,附赠热门例题
java·算法·图论
野犬寒鸦9 小时前
从零起步学习并发编程 || 第六章:ReentrantLock与synchronized 的辨析及运用
java·服务器·数据库·后端·学习·算法
霖霖总总9 小时前
[小技巧66]当自增主键耗尽:MySQL 主键溢出问题深度解析与雪花算法替代方案
mysql·算法
rainbow68899 小时前
深入解析C++STL:map与set底层奥秘
java·数据结构·算法