leetcode129. 求根节点到叶节点数字之和
给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。
每条从根节点到叶节点的路径都代表一个数字:
例如,从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。
计算从根节点到叶节点生成的 所有数字之和 。
叶节点 是指没有子节点的节点。
示例 1:
算法思想
使用深度优先搜索遍历整棵树,记录从根到当前节点的路径上的数字,当到达叶子节点时,将这个数字加到结果中。
算法具体步骤
1.定义一个全局变量 res 用于存储结果。
2.定义一个辅助函数 dfs,它接收当前节点和当前路径上的数字 temp 作为参数。
3.在 dfs 函数中,将当前节点的值加到 temp 上。
4.如果当前节点是叶子节点(既没有左子树也没有右子树),将 temp 加到 res 上。
5.否则,递归地在左子树和右子树上调用 dfs 函数。
算法代码
class Solution {
public:
int res=0;
void dfs(TreeNode* root,int temp)
{
temp=temp*10+root->val;
if(root->left==nullptr && root->right==nullptr)
{
res+=temp;
return ;
}
if(root->left)
{
dfs(root->left,temp);
}
if(root->right)
{
dfs(root->right,temp);
}
}
int sumNumbers(TreeNode* root) {
dfs(root,0);
return res;
}
};