通关剑指 Offer——剑指 Offer II 049. 从根节点到叶节点的路径数字之和

SK_Jaco

已于 2022-10-25 11:48:51 修改

阅读量178

点赞数 1

分类专栏： # 通关剑指Offer 文章标签： leetcode 算法数据结构 java

于 2022-10-25 11:47:16 首次发布

本文链接：https://blog.csdn.net/qq_38550836/article/details/127510628

版权

通关剑指Offer 专栏收录该内容

34 篇文章 0 订阅

订阅专栏

1.题目描述

剑指 Offer II 049. 从根节点到叶节点的路径数字之和

给定一个二叉树的根节点 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

2.解题思路与代码

2.1 解题思路

这是一道深度优先遍历的应用题，题目有两个要求，首先我们需要得到二叉树的每个路径从上到下遍历组成的数字，然后再将数字进行求和得到结果。首先我们使用过深度优先遍历整颗二叉树，当我们遇到叶节点时就将结果加上当前路径数字。那么如何得到二叉树路径组成的数字呢？我们使用变量 num 存放路径组成的数字，从根节点开始遍历，每遍历到一个节点就使用让 num 扩大十倍加上节点值即可，即 num = num*10+node.val。并且每次处理完子节点之后需要将 num 状态还原，即 num /= 10。

2.2 代码

class Solution {
    int ans = 0;
    int num = 0;

    public int sumNumbers(TreeNode root) {
        if (root == null) {
            return ans;
        }
        StringBuilder builder = new StringBuilder();
        process(root);
        return ans;
    }

    public void process(TreeNode node) {
        if (node.left == null && node.right == null) {
            num = num * 10 + node.val;
            ans += num;
            return;
        }
        num = num * 10 + node.val;
        if (node.left != null) {
            process(node.left);
            num /= 10;
        }
        if (node.right != null) {
            process(node.right);
            num /= 10;
        }
    }
}