路径总和

路径总和

CategoryDifficultyLikesDislikes
algorithmsEasy (53.55%)1222-
Tags

tree | depth-first-search

Companies

microsoft

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false

叶子节点 是指没有子节点的节点。

示例 1:

1
2
3
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true
解释:等于目标和的根节点到叶节点路径如上图所示。

示例 2:

1
2
3
4
5
6
输入:root = [1,2,3], targetSum = 5
输出:false
解释:树中存在两条根节点到叶子节点的路径:
(1 --> 2): 和为 3
(1 --> 3): 和为 4
不存在 sum = 5 的根节点到叶子节点的路径。

示例 3:

1
2
3
输入:root = [], targetSum = 0
输出:false
解释:由于树是空的,所以不存在根节点到叶子节点的路径。

提示:

  • 树中节点的数目在范围 [0, 5000]
  • -1000 <= Node.val <= 1000
  • -1000 <= targetSum <= 1000

Discussion | Solution

解法

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
// @lc code=start
// Definition for a binary tree node.
// #[derive(Debug, PartialEq, Eq)]
// pub struct TreeNode {
//   pub val: i32,
//   pub left: Option<Rc<RefCell<TreeNode>>>,
//   pub right: Option<Rc<RefCell<TreeNode>>>,
// }
//
// impl TreeNode {
//   #[inline]
//   pub fn new(val: i32) -> Self {
//     TreeNode {
//       val,
//       left: None,
//       right: None
//     }
//   }
// }
use std::cell::RefCell;
use std::rc::Rc;
impl Solution {
    /// ## 解题思路
    /// - 递归
    /// 1. 若节点为空,则遍历结束,未找到;
    /// 2. 若为叶子节点,且节点val==剩下的target, 则找到;
    /// 3. 否则递归在左右子树中查找target-node.val;
    pub fn has_path_sum(root: Option<Rc<RefCell<TreeNode>>>, target_sum: i32) -> bool {
        /// helper
        fn helper(node: &Option<Rc<RefCell<TreeNode>>>, target_sum: i32) -> bool {
            match node {
                None => false,
                Some(n)
                    if n.borrow().left.is_none()
                        && n.borrow().right.is_none()
                        && n.borrow().val == target_sum =>
                {
                    true
                }
                Some(n) => {
                    helper(&n.borrow().left, target_sum - n.borrow().val)
                        || helper(&n.borrow().right, target_sum - n.borrow().val)
                }
            }
        }

        helper(&root, target_sum)
    }
}
// @lc code=end

updatedupdated2024-08-252024-08-25