题目大意:给定一个二叉树和一个数值sum,问二叉树中是否存在一条从根节点到叶子节点的路径,使得这条路径所有节点的值之和等于sum。
理解分析:先序遍历二叉树,同时修改节点的值为本节点值加上其父节点的值。这样叶子节点存储的就是从根到叶子节点的所有节点值之和,每到一个叶子节点都进行判断,若相等,则返回true;直到所有节点都遍历过,不存在,则返回false。
实现:
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if(root == null) return false;
if(root.left != null) {
root.left.val += root.val;
boolean res = hasPathSum(root.left, sum);
if(res)
return true;
}
if(root.right != null) {
root.right.val += root.val;
boolean rr = hasPathSum(root.right, sum);
if(rr)
return true;
}
if(root.left == null && root.right == null && root.val == sum) return true;
return false;
}
}