[JavaScript / leetcode] 437. 路径总和 III

55 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第16天,点击查看活动详情

每日刷题 2022.10.14

题目

  • 给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
  • 路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。

示例

  • 示例1

image.png

输入:root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出:3
解释:和等于 8 的路径有 3 条,如图所示。
  • 示例2
输入: root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出: 3

提示

  • 二叉树的节点个数的范围是 [0,1000]
  • -10^9 <= Node.val <= 10^9 
  • -1000 <= targetSum <= 1000

解题思路

  • 所谓双递归,其实就是内外两层递归实现暴力遍历所有节点
  • 外层递归保证遍历到树的所有节点;内层递归才是真正寻找当前节点的递归.
  • 以当前节点为根节点,路径和符合要求的情况
  • 然后将所有情况加在一起就是最终的结果了。

复杂度分析

  • 外层递归遍历所有节点,内层节点遍历子树节点,所以最终是 1+2+3+...+N,所以平均O(N^2)
  • 空间复杂度:O(1)

AC代码

/**
 * 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) {
  // 使用递归的手法处理
  if(!root) return 0;
  // 设定初始值
  let result = 0, n = 0;
  // 调用函数
  bianli(root, [targetSum])
  return result;
  function bianli(root, tmp){
      if(!root) return
      n++;
      tmp = tmp.map(el => el-root.val)
      tmp.forEach(el => el==0?result += 1:null);
      if(n === 0) n = 0;
      if(root.left) bianli(root.left, [...tmp, targetSum])
      if(root.right) bianli(root.right, [...tmp, targetSum])
  }
};