题目描述
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ \
7 2 1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。
解法一:递归(DFS)
这里的关键从根到叶才判断路径和是否为sum
采用``if(root.leftnull&&root.rightnull)`
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if(root==null) return false;
//左右子树同时为空,才是叶子结点,此时的根节点是否为sum
if(root.left==null&&root.right==null) return root.val==sum;
//遍历左右子树将sum-根结点值
return hasPathSum(root.left,sum-root.val)||hasPathSum(root.right,sum-root.val);
}
}
解法二:BFS
public boolean hasPathSum(TreeNode root, int sum) {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
Queue<Integer> queueSum = new LinkedList<Integer>();
if (root == null)
return false;
queue.offer(root);
queueSum.offer(root.val);
while (!queue.isEmpty()) {
int levelNum = queue.size(); // 当前层元素的个数
for (int i = 0; i < levelNum; i++) {
TreeNode curNode = queue.poll();
int curSum = queueSum.poll();
if (curNode != null) {
//判断叶子节点是否满足了条件
if (curNode.left == null && curNode.right == null && curSum == sum) {
return true;
}
//当前节点和累计的和加入队列
if (curNode.left != null) {
queue.offer(curNode.left);
queueSum.offer(curSum + curNode.left.val);
}
if (curNode.right != null) {
queue.offer(curNode.right);
queueSum.offer(curSum + curNode.right.val);
}
}
}
}
return false;
}
解法三:栈
public boolean hasPathSum(TreeNode root, int sum) {
Stack<TreeNode> stack = new Stack<>();
Stack<Integer> stackSum = new Stack<>();
TreeNode cur = root;
int curSum = 0;
while (cur != null || !stack.isEmpty()) {
// 节点不为空一直压栈
while (cur != null) {
stack.push(cur);
curSum += cur.val;
stackSum.push(curSum);
cur = cur.left; // 考虑左子树
}
// 节点为空,就出栈
cur = stack.pop();
curSum = stackSum.pop();
//判断是否满足条件
if (curSum == sum && cur.left == null && cur.right == null) {
return true;
}
// 考虑右子树
cur = cur.right;
}
return false;
}