[路飞]_leetcode-面试题 04.12-求和路径

简介: leetcode-面试题 04.12-求和路径

网络异常,图片无法展示
|


「这是我参与2022首次更文挑战的第9天,活动详情查看:2022首次更文挑战


[题目地址][B站地址]


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


示例:


给定如下二叉树,以及目标和 sum = 22


5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
复制代码


返回:


3
解释: 和为 22 的路径有:[5,4,11,2], [5,8,4,5], [4,11,7]
复制代码


提示:


  • 节点总数 <= 10000


解题思路


本题要求我们在二叉树中,找出路径上节点和值等于目标值的路径的数量。这里的路径的起点可以是任何一个节点,终点也可以是任何一个节点,但是路径必须是从上到下的。


所以我们可以以任何一个节点为起点,向下查找路径,并记录节点和值,当和值等于目标值的时候,就说明我们找到了一个符合条件的路径,记录数量。这里递归函数的退出条件就是当我们找到一个空节点为止。


当我们处理完了这样的一个过程,就找到了所有符合条件的路径,也就得到了符合条件的路径的数量。


代码实现


var pathSum = function(root, sum) {
  // 获取路径 传入节点和目标值
  function getPath(node,target){
    // 如果当前节点为空,递归结束,返回 0
    // 因为空节点肯定是找不到等于目标值的路径的
    if(node===null) return 0;
    // 初始化数量为0,并记录当前节点值
    let num = 0,nodeVal = node.val
    // 如果当前节点值等于目标值,数量+1
    if(nodeVal === target) num++
    // 获取后续的目标值
    const val = target-nodeVal
    // 递归处理左子树
    num += getPath(node.left,val)
    // 递归处理右子树
    num += getPath(node.right,val)
    // 如果左子树不为空并且左子树没有被作为起点进行过查找
    // 以左子树为起点查找路径
    if(node.left&&!node.left.tag){
      // 给左子树打上标记,证明作为起点使用过
      node.left.tag = true;
      num += getPath(node.left,sum)
    }
    // 如果右子树不为空并且右子树没有被作为起点进行过查找
    // 以右子树为起点查找路径
    if(node.right&&!node.right.tag){
      // 给右子树打上标记,证明作为起点使用过
      node.right.tag = true;
      num += getPath(node.right,sum)
    }
    // 返回符合条件的路径的数量
    return num;
  }
  // 返回结果值
  return getPath(root,sum)
};
复制代码


这里要注意的是,因为递归过程中,有的节点会被经过多次,所以要记录当前节点是否被当做起点进行过查找,防止重复查找。


至此我们就完成了 leetcode-面试题 04.12-求和路径


如有任何问题或建议,欢迎留言讨论!👏🏻👏🏻👏🏻

相关文章
|
2月前
|
算法
【数组相关面试题】LeetCode试题
【数组相关面试题】LeetCode试题
|
2月前
力扣面试经典题之哈希表
力扣面试经典题之哈希表
18 0
|
2月前
|
存储
力扣面试经典题之数组/字符串
力扣面试经典题之数组/字符串
23 0
|
2月前
力扣面试经典题之二叉树
力扣面试经典题之二叉树
16 0
|
28天前
|
算法
【力扣经典面试题】121. 买卖股票的最佳时机
【力扣经典面试题】121. 买卖股票的最佳时机
|
28天前
|
存储
【力扣经典面试题】80. 删除有序数组中的重复项 II
【力扣经典面试题】80. 删除有序数组中的重复项 II
|
2月前
leetcode热题100.二叉树中的最大路径和
leetcode热题100.二叉树中的最大路径和
18 0
|
2月前
|
vr&ar
leetcode热题100.路径总和 III
leetcode热题100.路径总和 III
19 1
|
2月前
|
机器学习/深度学习 人工智能 算法
LeetCode刷题--- 面试题 01.07. 旋转矩阵(原地旋转+翻转替旋转)
LeetCode刷题--- 面试题 01.07. 旋转矩阵(原地旋转+翻转替旋转)
|
2月前
|
算法 测试技术 索引
力扣面试经典题之数组/字符串(二)
力扣面试经典题之数组/字符串(二)
13 0

热门文章

最新文章