面试算法50:向下的路径节点值之和

题目

给定一棵二叉树和一个值sum,求二叉树中节点值之和等于sum的路径的数目。路径的定义为二叉树中顺着指向子节点的指针向下移动所经过的节点,但不一定从根节点开始,也不一定到叶节点结束。例如,在如图8.5所示中的二叉树中有两条路径的节点值之和等于8,其中,第1条路径从节点5开始经过节点2到达节点1,第2条路径从节点2开始到节点6。

分析

虽然路径不一定从根节点开始,但仍然可以求得从根节点开始到达当前遍历节点的路径所经过的节点值之和。

如果在路径上移动时把所有累加的节点值之和都保存下来,然后移动的过程中求差值,就容易知道是否存在从任意节点出发的值为给定sum的路径。

有了前面的经验,就可以采用二叉树深度优先搜索来解决与路径相关的问题。当遍历到一个节点时,先累加从根节点开始的路径上的节点值之和,再计算到它的左右子节点的路径的节点值之和。这就是典型的前序遍历的顺序。

java 复制代码
public class Test {
    public static void main(String[] args) {
        TreeNode node5 = new TreeNode(5);
        TreeNode node2 = new TreeNode(2);
        TreeNode node4 = new TreeNode(4);
        TreeNode node1 = new TreeNode(1);
        TreeNode node6 = new TreeNode(6);
        TreeNode node3 = new TreeNode(3);
        TreeNode node7 = new TreeNode(7);

        node5.left = node2;
        node5.right = node4;
        node2.left = node1;
        node2.right = node6;
        node4.left = node3;
        node4.right = node7;

        int result = pathSum(node5, 8);
        System.out.println(result);
    }

    public static int pathSum(TreeNode root, int sum) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);// 节点和为0的路径有一个(空路径)

        // path: 遍历节点的路径和
        return dfs(root, sum, map, 0);
    }

    private static int dfs(TreeNode root, int sum, Map<Integer, Integer> map, int path) {
        if (root == null) {
            return 0;
        }

        // 前序遍历
        path += root.val;
        int count = map.getOrDefault(path - sum, 0);// 深度优先遍历,如果以前存在这个差值,那么和当前路径一定是以前路径的延伸
        map.put(path, map.getOrDefault(path, 0) + 1);

        count += dfs(root.left, sum, map, path);
        count += dfs(root.right, sum, map, path);

        // 当前这个节点遍历完成,重回当前节点的父节点继续遍历。
        map.put(path, map.get(path) - 1);

        return count;
    }
}
相关推荐
姚瑞南11 分钟前
【AI 风向标】四种深度学习算法(CNN、RNN、GAN、RL)的通俗解释
人工智能·深度学习·算法
渣哥22 分钟前
多环境配置利器:@Profile 在 Spring 项目中的实战价值
javascript·后端·面试
补三补四33 分钟前
SMOTE 算法详解:解决不平衡数据问题的有效工具
人工智能·算法
飞哥的AI笔记34 分钟前
热题解析:什么是Few-shot Learning?为什么给几个例子模型就能学会?
面试
RTC老炮1 小时前
webrtc弱网-RobustThroughputEstimator源码分析与算法原理
网络·算法·webrtc
南北是北北1 小时前
Handler线程模型与内存
面试
汤姆Tom1 小时前
CSS 新特性与未来趋势
前端·css·面试
南北是北北1 小时前
Thread ↔ Looper ↔ MessageQueue ↔ Handler ↔ Message之间的关系
面试
听风吹等浪起1 小时前
分类算法-逻辑回归
人工智能·算法·机器学习
敲代码的嘎仔1 小时前
JavaWeb零基础学习Day2——JS & Vue
java·开发语言·前端·javascript·数据结构·学习·算法