【算法设计与分析】求根节点到叶节点数字之和

本文介绍了一种使用递归方法计算二叉树中从根节点到叶节点路径生成的所有数字之和的算法,通过深度优先搜索遍历树结构,处理基本情况和递归子树,时间复杂度为O(N)。
摘要由CSDN通过智能技术生成

       📝个人主页:五敷有你      

 🔥系列专栏:算法分析与设计

⛺️稳中求进,晒太阳

题目

给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。

每条从根节点到叶节点的路径都代表一个数字:

  • 例如,从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。

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

叶节点 是指没有子节点的节点。

示例 

示例

示例 1:

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

示例 2:

输入:root = [4,9,0,5,1]
输出:1026
解释:从根到叶子节点路径 4->9->5 代表数字 495从根到叶子节点路径 4->9->1 代表数字 491从根到叶子节点路径 4->0 代表数字 40。因此,数字总和 = 495 + 491 + 40 = 1026

思路(递归)

        你可以使用深度优先搜索(DFS)来遍历树的所有路径。对于每个节点,你可以递归地计算从根节点到该节点的路径所代表的数字,并将其加到总和中。当遇到叶节点时,将路径所代表的数字加到总和中。

这里我们使用递归来解决这个问题。递归是一种解决问题的方法,其中函数调用自身来解决更小规模的问题。在树的情境下,递归特别适用,因为树的结构本身就是递归定义的。

递归函数的核心思路如下:

  1. 基本情况(终止条件)如果当前节点为空,则返回 0。
  2. 计算当前路径所代表的数字将当前路径的数字计算出来,即将当前的数字乘以 10 再加上当前节点的值。
  3. 处理叶子节点如果当前节点是叶子节点(即没有左右子节点),则返回当前路径所代表的数字。
  4. 递归左右子树对当前节点的左右子树分别调用递归函数,将当前路径所代表的数字传递给它们,并将它们返回的结果相加。
  5. 返回结果将左右子树返回的结果相加,得到从当前节点开始的所有路径所代表的数字之和。

我们就能够递归地计算出从根节点到每个叶子节点的路径所代表的数字之和。

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int sumNumbers(TreeNode root) {
       return sum(root,0);
    }
    public int sum(TreeNode root,int i){
        if(root==null) return 0;
        int temp=i*10+root.val;
        if(root.left==null&&root.right==null){
            return temp;
        }
        return sum(root.left,temp)+sum(root.right,temp);
    }
}

运行结果

时间复杂度为 O(N),其中 N 表示树的节点数量。

  • 37
    点赞
  • 34
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 10
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

五敷有你

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值