Leetcode刷题笔记 129. 求根到叶子节点数字之和

最新推荐文章于 2024-02-05 14:58:53 发布

jialun0116

最新推荐文章于 2024-02-05 14:58:53 发布

阅读量91

点赞数

分类专栏： leetcode 树文章标签： leetcode 二叉树

本文链接：https://blog.csdn.net/qq_30031221/article/details/109361871

版权

leetcode 同时被 2 个专栏收录

208 篇文章 0 订阅

订阅专栏

树

20 篇文章 0 订阅

订阅专栏

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

时间：2020年10月29日
知识点：递归、二叉树
题目链接：https://leetcode-cn.com/problems/sum-root-to-leaf-numbers/

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

例如，从根到叶子节点路径 1->2->3 代表数字 123。
计算从根到叶子节点生成的所有数字之和。
说明: 叶子节点是指没有子节点的节点。

示例 1:

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

示例 2:

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

思路

递归
退出条件：为空节点
加入答案条件：该节点没有子节点了
否则向左向右节点遍历

代码

#include <stdio.h>
#include <iostream>
using namespace std;
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
    int ans = 0;
    int sumNumbers(TreeNode* root) {
        sumNum(root, 0, ans);
        return ans;
    }
    void  sumNum(TreeNode*root,int now,int &ans){
        if(root == nullptr){
            return ;
        }
        else{
            now = now *10 +root->val;
            if(root->left)
               sumNum(root->left, now,ans);
            if(root->right)
               sumNum(root->right, now, ans);
            else if(root->left==nullptr && root->right==nullptr)
                ans += now;
        }
    }
};
int main()
{
    TreeNode root(1);
    TreeNode node2(2),node3(3);
    root.left = &node2;root.right = &node3;
    Solution s;
    cout<<s.sumNumbers(&root)<<endl;
    return 0;
}