- 关键字:树、深度优先搜索
- 难度:easy
- 题目大意:从给定的二叉树中,查找是否存在root->leaf路径和等于sum的路径。
题目:
Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.
Note: A leaf is a node with no children.
Example:
Given the below binary tree and sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ \
7 2 1
解题思路:
本题需要找到一条从root到leaf路径和为sum的路径,因此,我们可以使用深度优先去遍历每条路径,直至找到所求为止;
- 遍历一个节点时:
1、遍历到的节点为null时,直接返回false;
2、叶子节点时,判断此时路径和是否为sum;
3、不是叶子节点时,递归处理左右子树;
AC代码:
/**
* 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 -= root.val;
if(root.left==null&&root.right==null) {
return sum==0;
}
return hasPathSum(root.left,sum) || hasPathSum(root.right,sum);
}
}
复杂度:
使用深度搜索DFS,每个节点被访问一次。并且递归过程中,栈的最大深度为O(n)。考虑到本题并非平衡二叉树,最差将退化成链表,而大O代表复杂度的上阈值,因此为O(n)。
所以复杂度分析为:
- Time complexity: O(n)
- Space complexity: O(n)