牛客-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可以复习深浅拷贝。

全部评论

相关推荐

07-15 11:41
门头沟学院 Java
点赞 评论 收藏
分享
不对是145个人…嗯…&nbsp;大家都没发现秋招提前批来了嘛..笑死我了
牛客39712426...:投了也是浪费时间,之前投米实习,除了浪费我时间写笔试题没有任何反馈,懒得投了
26届校招投递进展
点赞 评论 收藏
分享
码砖:求职岗位要突出,一眼就能看到,教育背景放到最后,学校经历没那么重要,项目要重点突出
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务