leetcode 129. 求根到叶子节点数字之和

题目描述

给定一个二叉树,它的每个结点都存放一个0-9的数字,每条从根到叶子节点的路径都代表一个数字。
例如,从根到叶子节点路径 1->2->3代表数字 123
计算从根到叶子节点生成的所有数字之和。
说明: 叶子节点是指没有子节点的节点。
相关话题: 树、深度优先搜索    
难度:
 中等

示例1:

示例2:

思路:

  • 首先是先序遍历,每遍历到一个节点,如果该节点不是null,用传进来的sum * 10 + 当前节点的值sum = sum * 10 + root.val;当当前节点是叶子节点,返回sum,这就是一条路径的值。这是递归的子问题
  • 递归左、右子树,返回左右子树的值的和。
    return sumNumbers(root.left,sum) + sumNumbers(root.right,sum);
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int sumNumbers(TreeNode root) {
        return sumNumbers(root, 0);
    }
    public int sumNumbers(TreeNode root, int sum){
        if(root == null) return 0;
        sum = sum * 10 + root.val;
        if(root.left == null && root.right == null)
            return sum;
        return sumNumbers(root.left,sum) + sumNumbers(root.right,sum);
    }
}

总结:递归算法将大问题看小的思路实在是在奇妙了。

  • 在做递归的题时,有时候我们需要另起一个函数,然后在另一个函数中调用。例如本题另起了一个函数sumNumbers(TreeNode root, int sum)。因为在算一条路径的值时,是一种自顶向下的递归,必须接受上一步传递下来的结果,是一步一步更新的过程,所以需要sum的辅助。而在做斐波那契数列时,那是一个自底向上的递归,例如要算f(n),就要知道f(n-1),一直推到f(1),在此之前并没有结果算出来。
  • 二叉树的递归离不开左、右子树,根据实际问题需要确定子问题, 每个遍历到每个节点所需要做的操作(sum = sum * 10 + root.val),以及递归返回的条件(叶子节点时返回结果sum),然后就是借助左右思想return sumNumbers(root.left,sum) + sumNumbers(root.right,sum);
    做什么操作,什么时候返回
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 做题,实际写出例子,然后分析可能遇到的情况,慢慢的,思路就会出来了。 线性表 33. Search in Rota...
    小碧小琳阅读 1,612评论 0 2
  • 94. Binary Tree Inorder Traversal 中序 inorder:左节点->根节点->右节...
    Morphiaaa阅读 588评论 0 0
  • 一些概念 数据结构就是研究数据的逻辑结构和物理结构以及它们之间相互关系,并对这种结构定义相应的运算,而且确保经过这...
    Winterfell_Z阅读 5,994评论 0 13
  • 动态规划 111. 爬楼梯思路类似斐波那契数列注意考虑第 0 阶的特殊情况 272. 爬楼梯 II思路类似上题,只...
    6默默Welsh阅读 2,461评论 0 1
  • 一种底层思维方式 归纳演绎 解构重构图(线条,颜色,方向,权重……)
    龙猫同学阅读 595评论 0 0