题目描述:(链接)
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
.
解题思路:
深度优先搜索,遍历所有根节点到叶节点的路径。
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 return sumNumbers(root, 0); 14 } 15 private: 16 int sumNumbers(TreeNode *root, int sum) { 17 if (root == nullptr) return 0; 18 sum = sum * 10 + root->val; 19 if (root->left == nullptr && root->right == nullptr) { 20 return sum; 21 } 22 23 return sumNumbers(root->left, sum) + sumNumbers(root->right, sum); 24 } 25 };