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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值