难度:中等
给定一个二叉树,它的每个结点都存放一个 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
题目分析:
使用前序遍历(DFS),将上个节点值乘10+当前节点值,传入下次递归,直到叶节点为止,存入res中。
参考代码:
#include <iostream>
#include <vector>
#include <string>
#include <unordered_map>
#include <deque>
#include <stack>
#include <algorithm>
#include <map>
#include <assert.h>
#include <memory>
#include<queue>
#include<functional>
#include <set>
#include <unordered_set>
using namespace std;
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 == NULL)
return 0;
long long res = 0;
long long temp = 0;
sum_numbers(root,res,temp);
return res;
}
//前序遍历操作
void sum_numbers(TreeNode* root, long long& res, long long temp)
{
//递归的结束条件,当前节点为空
if (root == NULL)
return;
//判断是否走到叶子结点,根操作
if (root->left == NULL && root->right == NULL)
{
temp = temp * 10 + root->val;//当前节点值还未加,需要加上
res += temp;
return;
}
// 左 右
sum_numbers(root->left, res, temp*10 + root->val);
sum_numbers(root->right, res, temp * 10 + root->val);
}
};
int main()
{
Solution solution;
TreeNode TreeNode1 = TreeNode(4);
TreeNode TreeNode2 = TreeNode(9);
TreeNode TreeNode3 = TreeNode(0);
TreeNode TreeNode4 = TreeNode(5);
TreeNode TreeNode5 = TreeNode(1);
TreeNode1.left = &TreeNode2;
TreeNode1.right = &TreeNode3;
TreeNode2.left = &TreeNode4;
TreeNode2.right = &TreeNode5;
TreeNode3.left = NULL;
TreeNode3.right = NULL;
TreeNode4.left = NULL;
TreeNode4.right = NULL;
TreeNode5.left = NULL;
TreeNode5.right = NULL;
cout << solution.sumNumbers(&TreeNode1) << endl;
system("pause");
return 0;
}