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