題目描述

給定一個二叉樹，它的每個結點都存放一個 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.

解題思路

利用樹的前序遍歷思想，維護到當前節點的路徑值以及總路徑和，每次遍歷到一個節點首先更新當前路徑值，若遍歷到葉子節點就把路徑值添加到路徑總和中。

代碼

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     int sumNumbers(TreeNode* root) {
13         if(!root) return 0;
14         int sum = 0;
15         preOrder(root, sum, 0);
16         return
 
 sum;
17     }
18     void preOrder(TreeNode* root, int &sum, int num){
19         if(!root->left && !root->right)
20             sum += num * 10 + root->val;
21         else{
22             num = num * 10 + root->val;
23             if(root->left)
24                 preOrder(root->left, sum, num);
25             if(root->right)
26                 preOrder(root->right, sum, num);
27         }
28     }
29 };

LeetCode 129. 求根到葉子節點數字之和（Sum Root to Leaf Numbers）