leetcode-112-Path-Sum
题目描述(简单难度)
给定一个sum
,判断是否有一条从根节点到叶子节点的路径,该路径上所有数字的和等于sum
。
解法一 递归
这道题其实和 “>111 题 的分析,这道题无非是把 public boolean hasPathSum(TreeNode root, int sum) {
if (root == null) {
return false;
}
return hasPathSumHelper(root, sum);
}
private boolean hasPathSumHelper(TreeNode root, int sum) {
if (root == null) {
return sum == 0;
}
return hasPathSumHelper(root.left, sum - root.val) || hasPathSumHelper(root.right, sum - root.val);
}
看起来没什么问题,并且对于题目给的样例也是没问题的。但是对于下边的样例: 当某个子树只有一个孩子的时候,就会出问题了,可以看 public boolean hasPathSum(TreeNode root, int sum) {
if (root == null) {
return false;
}
return hasPathSumHelper(root, sum);
}
private boolean hasPathSumHelper(TreeNode root, int sum) {
//到达叶子节点
if (root.left == null && root.right == null) {
return root.val == sum;
}
//左孩子为 null
if (root.left == null) {
return hasPathSumHelper(root.right, sum - root.val);
}
//右孩子为 null
if (root.right == null) {
return hasPathSumHelper(root.left, sum - root.val);
}
return hasPathSumHelper(root.left, sum - root.val) || hasPathSumHelper(root.right, sum - root.val);
}
同样的,我们可以利用一个队列对二叉树进行层次遍历。同时还需要一个队列,保存当前从根节点到当前节点已经累加的和。 解法一其实本质上就是做了 3
/ \
9 20
/ / \
8 15 7
sum = 12
解法二 BFS
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;
}
解法三 DFS
DFS
,我们知道DFS
可以用栈去模拟。对于这道题,我们可以像解法二的BFS
一样,再增加一个栈,去保存从根节点到当前节点累计的和就可以了。
看完两件小事
如果你觉得这篇文章对你挺有启发,我想请你帮我两个小忙:
- 把这篇文章分享给你的朋友 / 交流群,让更多的人看到,一起进步,一起成长!
- 关注公众号 「方志朋」,公众号后台回复「666」 免费领取我精心整理的进阶资源教程
本文著作权归作者所有,如若转载,请注明出处
转载请注明:文章转载自「 Java极客技术学习 」https://www.javajike.com