leetcode437.路径总和III

leetcode437.路径总和III

题目

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

用例

输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出：3
解释：和等于 8 的路径有 3 条，如图所示。

输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：3

求解

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} targetSum
 * @return {number}
 */
var pathSum = function(root, targetSum) {
    let res=0
    pre(root)
    
    function pre(root){
        if(root==null){
            return
        }
        sum(root,targetSum)
        pre(root.left)
        pre(root.right)
    }

    function sum(root,a){
        if(root==null){
            return
        }
        a=a-root.val
        if(a==0){
            res++
        }
        sum(root.left,a)
        sum(root.right,a)
    }
    return res
};