求二叉树中以元素x为根的子树的深度C语言代码
时间: 2024-05-06 18:20:34 浏览: 53
下面是一个示例代码:
```
int subtreeDepth(struct TreeNode* root, int x) {
if (root == NULL) {
return 0;
}
if (root->val == x) {
return maxDepth(root);
}
int leftDepth = subtreeDepth(root->left, x);
int rightDepth = subtreeDepth(root->right, x);
return max(leftDepth, rightDepth);
}
int maxDepth(struct TreeNode* root) {
if (root == NULL) {
return 0;
}
int leftDepth = maxDepth(root->left);
int rightDepth = maxDepth(root->right);
return max(leftDepth, rightDepth) + 1;
}
```
这里使用了两个函数,`subtreeDepth`用于求以元素x为根的子树的深度,`maxDepth`用于求一棵二叉树的最大深度。`subtreeDepth`函数首先判断当前节点是否为x,如果是,则返回该节点的深度,否则递归调用左右子树,返回左右子树深度的最大值。`maxDepth`函数则是标准的求二叉树深度的递归函数。
阅读全文