C语言实现给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。 每条从根节点到叶节点的路径都代表一个数字: 例如,从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。 计算从根节点
时间: 2024-03-09 22:46:53 浏览: 91
到每个叶节点的路径所代表的数字之和,然后返回所有路径数字之和的总和。你可以使用深度优先搜索算法(DFS)来遍历整个二叉树,计算每个叶节点所表示的数字,并将其加入到总和中。
以下是使用 C 语言实现的代码:
```c
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int sumNumbers(struct TreeNode* root) {
if (root == NULL) {
return 0;
}
int totalSum = 0;
dfs(root, 0, &totalSum);
return totalSum;
}
void dfs(struct TreeNode* node, int pathSum, int* totalSum) {
pathSum = pathSum * 10 + node->val;
if (node->left == NULL && node->right == NULL) {
*totalSum += pathSum;
}
if (node->left != NULL) {
dfs(node->left, pathSum, totalSum);
}
if (node->right != NULL) {
dfs(node->right, pathSum, totalSum);
}
}
```
在上面的代码中,我们定义了一个 `sumNumbers` 函数,它接收一个二叉树的根节点作为参数,并返回所有路径数字之和的总和。在 `sumNumbers` 函数中,我们首先判断根节点是否为空,如果为空,则直接返回 0。
接下来,我们定义了一个 `totalSum` 变量,用于记录路径数字之和的总和。我们通过调用 `dfs` 函数来计算每个叶节点所表示的数字,并将其加入到 `totalSum` 中。
在 `dfs` 函数中,我们首先将当前节点的值加入到 `pathSum` 变量中,并将 `pathSum` 乘以 10,以便计算下一个节点的路径数字。如果当前节点是叶节点,则将 `pathSum` 加入到 `totalSum` 中。
最后,我们递归遍历左右子树,分别调用 `dfs` 函数,并将 `pathSum` 和 `totalSum` 作为参数传递进去。
这样,我们就可以使用上述代码来计算二叉树中所有路径数字之和的总和。
阅读全文