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

题目描述
给定一棵二叉树,仅包含 0-9,每条从根节点到叶节点的路径表示一个整数。
例如,如果根节点到叶节点的路径是1->2->3,表示的数就是123。

请计算所有从根节点到叶节点的路径表示的数的和。

注意;叶节点指不包含儿子的节点。

样例1
输入:

[1,2,3]
    1
   / \
  2   3

输出:

25

解释:
总共有两条根到叶的路径:1->2和1->3,表示的数分别是12和13,所以总和是25。

样例2
输入:

[4,9,0,5,1]
    4
   / \
  9   0
 / \
5   1

输出:

1026

解释:总共存在三条根到叶的路径:4->9->5, 4->9>1, 4->0,表示的数分别是495,491,40,总和是1026。

算法
(树的遍历) O(n)
从根节点递归遍历整棵树,遍历时维护从根节点到该节点的路径表示的数,当遍历到叶节点时,将路径表示的数累加到答案中。

时间复杂度分析:每个节点仅被遍历一遍,所以时间复杂度是O(n)。

思路 :

  • 定义一个ans记录答案
  • 深搜的过程当中,维护一个number,每次都经过一个number进行更新! 截至条件 :
  • 当前节点为叶子节点(左右子树为空) ,将此刻number累加到ans
  • 返回 ans即可
class Solution {
    
    
public:
    int ans = 0 ;
    int sumNumbers(TreeNode* root) {
    
    
        if(root) dfs(root, 0) ;
        return ans ;
    }
    void dfs(TreeNode* root , int num){
    
    
        num = num * 10 + root -> val ;
        if(!root -> left && !root -> right) ans += num ;
        if(root -> left) dfs(root -> left , num) ;
        if(root -> right) dfs(root -> right ,num) ;
    }
};

Guess you like

Origin blog.csdn.net/m0_46571920/article/details/120983783