LeetCode之求根到叶子节点数字之和

最新推荐文章于 2023-09-02 18:47:36 发布

原创最新推荐文章于 2023-09-02 18:47:36 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一道关于二叉树的算法题，题目要求计算所有从根节点到叶子节点路径上的数字之和。通过深度优先遍历的方法，递归地计算每个路径的数字，并在遇到叶子节点时累加到总和中。

求根到叶子节点数字之和

题目

给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。

例如，从根到叶子节点路径 1->2->3 代表数字 123。

计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

示例 1:

输入: [1,2,3]
1
/ \
2 3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此，数字总和 = 12 + 13 = 25.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-root-to-leaf-numbers

思路

1.因为是到叶子节点所以基本思路是深度优先遍历。
2.首先，把第一层节点对于的值先求出来。再求下一层时只要对上一次的值*10+Node.val就可以得出这个节点对应的值。
3.分为左右两个部分递归求解。
4.程序的出口是叶子节点也就是Node.left和Node.right都是null。

代码

/**
 * 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) {
        if(root == null){
            return 0;
        }
        return sumAll(root,0);
    }
    public int sumAll(TreeNode Node,int lastSum){
        if(Node == null){
            return 0;
        }
        int sum = lastSum*10 + Node.val;
        if(Node.left == null && Node.right == null){
            return sum;
        }
        return sumAll(Node.left,sum) + sumAll(Node.right,sum);
    }
}