题目描述
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
示例
递归
这道题主要考察的是对二叉树的遍历。而二叉树的遍历可以使用层序遍历,这种方式需要用到队列作为辅助的数据结构。另一种方式则是递归,递归使用的是JVM的栈,方便许多。每次经过一个节点,都要使得sum减去此节点的值,表示记录下了此节点对于sum的贡献。我们对于每个叶子节点(注意是叶子结点,左右子树均为空,本人一开始就看错了,以为只要是子节点就行,结果很悲剧)若此时的sum=0,则意味着这个叶子结点到root结点的和恰好为sum。否则继续查找子树里是否存在使得和与sum相等的叶子节点。只要左右子树有一个存在叶子结点使得sum=0,就返回true。
public boolean hasPathSum(TreeNode root, int sum) {
if (root == null)
return false;
sum = sum - root.val;//表示将此节点计入对sum的贡献。
if (root.right == null && root.left == null)//判断此叶子结点是否符合要求
return sum == 0;
boolean leftSumResult = false, rightSumResult = false;
//查找左右子树
leftSumResult = hasPathSum(root.left, sum);
rightSumResult = hasPathSum(root.right, sum);
return leftSumResult || rightSumResult;
}
这段代码的效率还算高,执行时间击败了100%的用户。