112. Path Sum
题干简述:
给我们一个二叉树的根节点与一个目标值,判断当前二叉树有没有一条由根节点到叶子节点组成的路径,该路径上所有节点的值的和等于目标值。
叶子节点,即没有子节点的节点。
解题思路
这一题比较新颖的一点在于,它是计算一条路径上所有节点的和。为了达到这个目的,DFS 后序优先遍历就不适用了,我们总不能从最底层的叶子节点开始,计算每一条路径和,去和目标值进行比较吧。DFS 递归的思想,这里依旧可以正常使用,只不过我们需要从上到下,一步步的来算,也就是通过前序遍历来解决这个问题。
首先,递归解法的话,我们需要思考,如何让这个递归函数结束,并且返回一个有效可用的结果。结合题干,当我们传入递归函数的节点为 null 时,这时我们自然要 return false
,毋庸置疑。当我们传入的节点为叶子节点的时候,我们直接将其与此时的目标值进行对比,当与目标值一致时,return false
即可。注意,这里我们提到了此时的目标值 。在递归过程中,我们每经过一个节点,在这个节点不满足时,进入下一轮递归时,我们需要把目标值减去这个节点的值,传入后序的递归函数中,这样,就能保证我们在执行到最后一步时,可以正确的对比目标值与节点值。当当前节点非叶子节点,且不满足和为target时,我们就来到了一个分支,我们如何处理左右子树呢?答案很简单,都做处理,只要左右子树有一个满足条件,整个树就满足条件了。
题解
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @param {number} targetSum
* @return {boolean}
*/
var hasPathSum = function(root, targetSum) {
if (!root) return false;
if (!root.left && !root.right && root.val === targetSum) return true;
return hasPathSum(root.left, targetSum - root.val) || hasPathSum(root.right, targetSum - root.val)
};
BFS 广度优先遍历的解法其实也差不多,通过数组存储每层节点的同时,需要那一个变量存储当前路径的和,每次执行出队的时候,比较这个变量与 targetSum 的关系,判断是否满足要求即可。
Q.E.D.