LeetCode Path Sum

来源：互联网发布：淘宝怎么弄天猫积分编辑：程序博客网时间：2024/05/24 02:49

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.

For example:
Given the below binary tree and sum = 22,

              5             / \            4   8           /   / \          11  13  4         /  \      \        7    2      1

return true, as there exist a root-to-leaf path 5->4->11->2 which sum is 22.

思路分析：这题看到立刻就可以想到可以用DFS解，但是实现的时候有点小技巧，就是判断等于target的时机，并不是DFS搜索的每一步都判断，而是当搜索到叶子节点的时候判断是否等于target，还有如果当前节点搜索失败（已经是叶子节点但是当前和不是target），程序会回溯到父节点，继续搜索其他子树，注意当回溯的时候我们不需要对curSum减去子节点的val来恢复“现场”，原因是基础数据类型（int，char，……）传值，对象类型（Object，数组，容器……）传引用，所以递归调用并没有修改curSum的值，我们只是传入了一份curSum的拷贝。总体是DFS实现，但是和DFS还是有区别的，因为是找root到leaf的path和，而不是整棵树所有node求和。时间复杂度O（n），相当于扫描了一遍所有node，空间复杂度考虑到递归栈调用需要O（logn）。

AC Code

/** * Definition for binary tree * public class TreeNode { *     int val; *     TreeNode left; *     TreeNode right; *     TreeNode(int x) { val = x; } * } */public class Solution {    public boolean hasPathSum(TreeNode root, int sum) {        return helper(root, sum, 0);    }        public boolean helper(TreeNode root, int target, int curSum){        if(root == null) return false;        curSum += root.val;        if(root.left == null && root.right ==  null && curSum == target) return true;        return helper(root.left, target, curSum) || helper(root.right, target, curSum);    }}

0 0