描述
给定一个二叉树root和一个值 sum ,判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点,不能从子节点到父节点
4.总节点数目为n.
思路
按照前序遍历遍历接点,把每个节点的值都添加到count中,当走到叶子节点时,如果count == sum就返回true,否则就返回到父节点,并且将count的值减去刚刚加入进去的值。
代码
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
/**
*
* @param root TreeNode类
* @param sum int整型
* @return bool布尔型
*/
boolean res = false;
public boolean hasPathSum (TreeNode root, int sum) {
if(sum == 0 || root == null){
return false;
}
func(root,0,sum);
return res;
}
public void func(TreeNode root, int count,int sum){
if(res){
return;
}
count = count + root.val;
if(root.left != null && count != sum){
func (root.left,count,sum);
}
if(root.right != null && count != sum){
func (root.right,count,sum);
}
if(count == sum && root.left == null && root.right == null){
//标致位,说明该路径已经找到了。
res = true;
return;
}else{
//走到叶子节点后,没有找到就结束,返回到父节点。
return;
}
}
}