给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ \
7 2 1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。
1.递归
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if(root==null)
return false;
sum=sum-root.val;
if(root.left==null&&root.right==null)
return (sum==0);
return hasPathSum(root.left,sum)||hasPathSum(root.right,sum);
}
2.循环
1.广度优先搜索
2.使用node栈,sum栈保存节点与当前sum
3.当node栈不为空的时候,遍历出栈。
4.当当前node左右子树都为空并且当前sum==0 return true;否则当前节点右子树入栈,sum入栈;左子树入栈,sum入栈。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if(root==null)
return false;
LinkedList<TreeNode> node_stack = new LinkedList();//保存node
LinkedList<Integer> sum_stack = new LinkedList();//当前sum
node_stack.add(root);
sum_stack.add(sum-root.val);
TreeNode node;//存放当前node
int cur_sum;
while(!node_stack.isEmpty()){//遍历node栈
node=node_stack.pollLast();
cur_sum=sum_stack.pollLast();
if((node.left==null)&&(node.right==null)&&(cur_sum==0))//遍历至叶子节点判断sum
return true;
if(node.right!=null){//不是叶子节点 继续深度遍历
node_stack.add(node.right);
sum_stack.add(cur_sum-node.right.val);
}
if(node.left!=null){
node_stack.add(node.left);
sum_stack.add(cur_sum-node.left.val);
}
}
return false;
}
}