面试题 04.12. 求和路径

Posted 2021-10-23 不吐西瓜籽

算法记录

LeetCode 题目：

  给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。

---

思路

• 算法记录
• 说明
• • 一、题目
  • 二、分析
• 总结

---

说明

一、题目

  设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

二、分析

• 相当于一个递归求和的问题，分别对当前节点和其左右节点进行递归的向下求解，返回三者的结果之和即可。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int pathSum(TreeNode root, int sum) {
        if(root == null) return 0;
        int l = pathSum(root.left, sum);
        int r = pathSum(root.right, sum);
        return l + r + getSum(root, sum);
    }

    private int getSum(TreeNode root, int sum) {
        if(root == null) return 0;
        sum -= root.val;
        return (sum == 0 ? 1 : 0) + getSum(root.left, sum) + getSum(root.right, sum);
    }
}

---

总结

熟悉二叉树的递归遍历方法。