牛客-NC8-二叉树根节点到叶子节点和为指定值的路径

NC8. 二叉树根节点到叶子节点和为指定值的路径(medium)


方法一:DFS法

思路:前序遍历的增强版,要找出根到叶节点的节点值之和等于sum的路径,所以dfs方法需要带一些标记数据,这里包括包含当前节点时的cur_sum和将当前节点纳入考虑的select列表。(注意这里有个Java常问的知识点:深浅拷贝,我被这个东西坑了,一直得不到正确的答案,这里必须对select列表进行深拷贝,因为浅拷贝生成的列表会指向之前的列表,后续对该列表的改变都会映射到之前的列表中导致最终达到目标叶子节点会将之前遍历得到的节点全记录着。)

import java.util.*;

/* * public class TreeNode { * int val = 0; * TreeNode left = null; * TreeNode right = null; * } */

public class Solution {
   
    /** * * @param root TreeNode类 * @param sum int整型 * @return int整型ArrayList<ArrayList<>> */
    ArrayList<ArrayList<Integer>> ret = new ArrayList<>();
    public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
   
        // write code here
        if (root == null) return ret;
        dfs(root, sum, new ArrayList<Integer>());
        
        return ret;
    }
    public void dfs(TreeNode node, int cur_sum, ArrayList<Integer> select) {
   
        
        ArrayList<Integer> newSelect = new ArrayList<>(select);
        newSelect.add(node.val);
        cur_sum -= node.val;
        
        // 到达叶节点
        if (node.left == null && node.right == null) {
   
            if (cur_sum == 0) {
   
                ret.add(newSelect);              
            } 

        }
        
        if (node.left != null) {
   
            dfs(node.left, cur_sum, newSelect);
        }
        if (node.right != null) {
   
            dfs(node.right, cur_sum, newSelect);
        }
    }
}

时间复杂度: O(N),考虑最坏情况下,二叉树变成一个长链结构,此时复杂度最高。
空间复杂度: O(N),需要考虑系统递归栈,最坏情况下,全部节点需要入栈。
总结:这道题复习了浅拷贝和深拷贝,这里分享给大家一个link可以复习深浅拷贝。

全部评论

相关推荐

01-23 14:54
同济大学 Java
热爱敲代码的程序媛:给你提几点【专业技能】这个模块里面可优化的地方:1.【具备JVM调优经验】可以去b站上搜一下JVM调优的视频,估计一两个小时凭你的学习能力就能掌握JVM调优的实践方面的技能。2.【MySql优化】MySql这一栏,你去b站或者找个博客看看MySql优化,学一下,如果你本身比较熟悉MySql语句的话,那基本半天时间凭你的学习能力MySql语句优化方面的技能你也能掌握个差不多。以上1,2两点主要是因为我看你专业技能大部分都说的是偏理论,没有写应用。再就是最后,你结合你的项目,想一想你的项目中哪些sql语句是可以用MySql优化的,到时候你面试的时候也好结合着说一下。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务