LeetCode112-路径总和

发表于 | 分类于

题目链接

英文链接：https://leetcode.com/problems/path-sum/

中文链接：https://leetcode-cn.com/problems/path-sum/

题目详述

给定一个二叉树和一个目标和，判断该树中是否存在根节点到叶子节点的路径，这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树，以及目标和 sum = 22，

1
2
3
4
5
6
7
      5
     / \
    4   8
   /   / \
  11  13  4
 /  \      \
7    2      1

返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

题目详解

方法一：递归。

  • root 为 null 直接返回 false。
  • 到达叶子结点判断是否满足条件。
  • 非叶子节点继续往下找。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
public class LeetCode_00112 {

    // 递归
    public boolean hasPathSum(TreeNode root, int sum) {
        if (root == null) {
            return false;
        }
        sum -= root.val;
        if (root.left == null && root.right == null) {
            return sum == 0;
        }
        return hasPathSum(root.left, sum) || hasPathSum(root.right, sum);
    }
}

方法二：迭代。

  • 将递归改为迭代也很容易，不过会改变给定树中结点的值。
  • 推荐使用递归，容易理解且无副作用。
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
public class LeetCode_00112 {

    // 迭代
    public boolean hasPathSum(TreeNode root, int sum) {
        if (root == null) {
            return false;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            if (node.left == null && node.right == null && node.val == sum) {
                return true;
            }
            if (node.right != null) {
                node.right.val += node.val;
                stack.push(node.right);
            }
            if (node.left != null) {
                node.left.val += node.val;
                stack.push(node.left);
            }
        }
        return false;
    }
}