题目要求:
这个题目说的是,给你一棵二叉树和一个整数,你要判断这棵二叉树上是否存在一条从根到叶子节点的路径,这条路径上所有节点中的数字相加等于给你的整数。
比如说,给你的二叉树是:
1
/
2 4
/
8 16
给你的整数是 13。在这棵二叉树中存在一条从根到叶子节点的路径 1->4->8,路径上的数字加起来等于 13,于是要返回 true。
思路:
二叉树的问题 都可以用递归的方式去做,遍历左子树和右子树
主要是边界条件的判断, 这里要求根到叶子结点的路径, 那么最后左右结点一定为空,
sum之和 可以变成一路子上减掉当前结点的值
递归的代码量只有三行,比较简单,方法如下:
public boolean hasSumRecursive(TreeNode root,int sum){
if(root==null) return false;
if(root.leftnode==null &&root.rightnode==null) return sum==root.val;
return hasSumRecursive(root.leftnode,sum-root.val) ||
hasSumRecursive(root.rightnode,sum-root.val);
}
所以采用第二种方式 迭代的方式一步一步来写
而二叉树的迭代方法一般都是用一个栈来存储数据以方便记录路径
总结起来如下:
1、如果只有根节点或者找到叶子节点,我们就把其对应的val值返回
2、如果不是叶子节点,我们分别对根节点的左子树、右子树进行递归,直到找到叶子结点。然后遍历把叶子结点和父节点对应的val组成的序列返回上一层;
代码
import java.util.Stack;
public class PathSum62 {
public class TreeNode{
int val;
TreeNode leftnode;
TreeNode rightnode;
TreeNode(int x){
val=x;
}
}
public boolean hasPathSum(TreeNode root,int sum){
if(root==null){
return false;
}
Stack<TreeNode> stack=new Stack<>();
Stack<Integer> num=new Stack<>();
stack.push(root);
num.push(sum);
while (!stack.isEmpty()){
TreeNode n=stack.pop();
int s=num.pop();
if(n.leftnode==null &&n.rightnode==null&&n.val==s) return true;
if(n.leftnode!=null){
stack.push(n.leftnode);
num.push(s-n.val);
}
if(n.rightnode!=null){
stack.push(n.rightnode);
num.push(s-n.val);
}
}
return false;
}