首页 > 代码库 > LeetCode: Sum Root to Leaf Numbers [129]

LeetCode: Sum Root to Leaf Numbers [129]

2024-07-09 11:28:57   223人阅读

【题目】

Given a binary tree containing digits from 0-9 only, each root-to-leaf path could represent a number.

An example is the root-to-leaf path 1->2->3 which represents the number 123.

Find the total sum of all root-to-leaf numbers.

For example,

    1
   /   2   3

The root-to-leaf path 1->2 represents the number 12.
The root-to-leaf path 1->3 represents the number 13.

Return the sum = 12 + 13 = 25.


【题意】

    给定一棵二叉树，节点值只可能是[0-9]区间上的值，每一条从根到叶子的节点都可以看成一个整数，现要求把所有路径表示的整数相加，返回和


【思路】

    DFS，找到左右的路径，实数化每条路径上组合数。将所有的路径上得到的整数求和。
    
    这里2有个问题：
    1. 如果某条路径太长，组合数已经超出了int的上界怎么办
    2. 如果最终的和超出了int的上界怎么办
    
    题目没有进一步的说明，我们默认所给的测试用例都保证不会越界。

    


【代码】

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    
    void dfs(int&result, int num, TreeNode*node){
        //result表示所有路径的组合
        //num表示根到node的父节点的组合数
        if(node){
            num=10*num+node->val;   //计算从根到当前节点的组合数
            if(node->left==NULL && node->right==NULL){
                result+=num;    //已经找到一个条路径的组合数，累加到result上
                return;
            }
            
            if(node->left) dfs(result, num, node->left);
            if(node->right) dfs(result, num, node->right);
        }
    }


    int sumNumbers(TreeNode *root){
        if(root==NULL)return NULL;
        int result=0;
        int num=0;
        dfs(result, num, root);
        return result;
    }
};


联系
我们