题目描述

 

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