112&113. Path Sum & Path Sum II

Path 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.

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.
最先想到的就是使用递归了,我的方法使用了一个全局变量来记录整棵树中是否存在这样一个路径等于给出的数字。但是这个算法无论如何都会遍历整个二叉树,其实我们只要找到一条路符合要求就可以退出了。

var hasPathSum = function(root, sum) {
    //递归1
    if (!root) {
        return false;
    }
    var result = false;
    var help = function(root,sumRecord) {
        var sumNow = sumRecord+root.val;
        if (root.right) 
            help(root.right,sumNow);
        if (root.left) 
            help(root.left,sumNow);
        if (!root.left&&!root.right) {
            if (sumNow===sum)
                result = result || true;
            else 
                result = result || false;
        }
    };
    help(root,0);
    return result;
}

第二个递归算法比较简洁,直接利用给出的sum,在叶子节点减到0就意味着符合要求。这里就不用使用help函数了,直接递归就好,比较简洁,但是其实还是遍历了整棵树,本质上没有区别。

var hasPathSum = function(root, sum) {
    if (root === null) return false;
    sum = sum - root.val;
    if (root.left === null && root.right === null && sum === 0) return true;
    return hasPathSum(root.left, sum) || hasPathSum(root.right, sum);
};

使用递归我想不到怎么可以不遍历整棵树,于是只好换一种思路,使用深度优先遍历来进行每次从栈顶取出一个节点的时候,就将这个节点的值加到它的左右子节点中,并把左右节点入栈,如果没有子节点,就将当前节点的值与要求的总和做对比,符合就直接返回true。

var hasPathSum = function(root, sum) {
    if (!root) {
        return false;
    }
    var stack = [root];
    while (stack.length!==0) {
        var node = stack.pop();
        if (node.left) {
            node.left.val += node.val;
            stack.push(node.left);
        }
        if (node.right) {
            node.right.val += node.val;
            stack.push(node.right);
        }
        if (!node.left&&!node.right) {
            if (node.val===sum)
                return true;
        }
    }
    return false;
};

Path Sum II

Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.
For example:Given the below binary tree and sum = 22,

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

return

[
   [5,4,11,2],
   [5,8,4,5]
]

这里其实和上面的思路一样,就是多一个path的记录

var pathSum = function(root, sum) {
    var res = [];
    var help = function(root,sumNow,path) {
        sumNow = sumNow + root.val;
        var pathNow = path.concat();
        pathNow.push(root.val);
        if (root.left)
            help(root.left, sumNow, pathNow);
        if (root.right)
            help(root.right, sumNow, pathNow);
        if (!root.left&&!root.right&&sumNow===sum) 
            res.push(pathNow);
    };
    if (root!==null)
        help(root,0,[]);
    return res;
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,768评论 0 33
  • 目录 简书的 markdown 都不支持 [TOC] 语法……我就不贴目录了。下面按照类别,列出了29道关于二叉树...
    被称为L的男人阅读 3,369评论 0 8
  • 在我看来,电影也是有衣服的。 一般的电影是穿着各式各样风格的外衣,文艺的、魔幻的、朴实或者华丽的;A片是脱去了外衣...
    一只煎蛋阅读 2,788评论 3 15
  • 缆车也是第一次乘坐,其实已经经历了激烈震荡的过山车,缆车应该算是温和许多。但是于我而言,或许这还是恐惧。等待的过程...
    微凉的海水和风阅读 219评论 0 0
  • 最近天气骤冷,超级冷的那种。 周末,美美的睡了一觉,刚醒,躺在暖暖的被窝里码字。 在这干冷干冷的日子里,却充斥着特...
    青柠__阅读 584评论 3 5