LeetCode 129.Sum Root to Leaf Numbers (求根到叶子节点数字之和)

Love The Way You Lie 2022-05-08 06:04 159阅读 0赞

题目描述：

给定一个二叉树，它的每个结点都存放一个 `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.

AC C++ Solution:

DFS算法：每到达叶节点，进行累加

/**
     * Definition for a binary tree node.
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution {
    public:
        int sumNumbers(TreeNode* root) {
            if(!root)
                return 0;
            
            int currentSum = 0;
            sum = 0;
            DFS(root,currentSum);
            return sum;
        }
        
        void DFS(TreeNode *node, int currentSum) {
            currentSum = currentSum*10 + node->val;
            if(!node->left && !node->right)
                sum += currentSum;
            
            if(node->left)
                DFS(node->left,currentSum);
            if(node->right)
                DFS(node->right,currentSum);
        }
    
    private:
        int sum;    
    };

--------------------