以下是一个生成求指定节点在二叉排序树中所在层次的 C 语言代码示例:
```c
#include
#include
// 定义二叉树节点
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
};
// 创建二叉树节点
struct TreeNode* createNode(int val) {
struct TreeNode* newNode = (struct TreeNode*)malloc(sizeof(struct TreeNode));
newNode->val = val;
newNode->left = NULL;
newNode->right = NULL;
return newNode;
}
// 向二叉排序树插入节点
struct TreeNode* insertNode(struct TreeNode* root, int val) {
if (root == NULL) {
return createNode(val);
}
if (val < root->val) {
root->left = insertNode(root->left, val);
} else if (val > root->val) {
root->right = insertNode(root->right, val);
}
return root;
}
// 求指定节点在二叉排序树中所在层次
int getNodeLevel(struct TreeNode* root, int val, int level) {
if (root == NULL) {
return 0; // 节点不存在,返回 0
}
if (val == root->val) {
return level; // 找到节点,返回当前层次
} else if (val < root->val) {
return getNodeLevel(root->left, val, level + 1); // 在左子树中递归查找
} else {
return getNodeLevel(root->right, val, level + 1); // 在右子树中递归查找
}
}
int main() {
// 创建二叉排序树
struct TreeNode* root = NULL;
root = insertNode(root, 4);
insertNode(root, 2);
insertNode(root, 1);
insertNode(root, 3);
insertNode(root, 6);
insertNode(root, 5);
insertNode(root, 7);
// 指定节点的值
int targetValue = 3;
// 求指定节点在二叉排序树中所在层次
int level = getNodeLevel(root, targetValue, 1);
if (level > 0) {
printf("节点 %d 在二叉排序树中的层次为 %d\n", targetValue, level);
} else {
printf("节点 %d 不存在于二叉排序树中\n", targetValue);
}
// 释放二叉树的内存
// 这里省略了释放内存的代码,实际使用中应该释放二叉树的内存
return 0;
}
```
这段代码定义了一个 `TreeNode` 结构体表示二叉树节点,以及相应的创建节点和插入节点的函数。然后,我们定义了一个 `getNodeLevel` 函数,它接受二叉排序树的根节点 `root`、目标节点的值 `val` 和当前层次 `level`,并递归查找目标节点在二叉排序树中所在的层次。
在 `main` 函数中,我们创建了一个二叉排序树,并指定了目标节点的值 `targetValue`。然后,我们调用 `getNodeLevel` 函数来求目标节点在二叉排序树中的层次,并根据结果打印相应的信息。
请注意,这只是一个示例代码,你可以根据自己的需求进行修改和扩展。另外,为了完整性和正确性,请确保在使用完二叉树后释放其内存以避免内存泄漏。
#include
#include
// 定义二叉树节点
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
};
// 创建二叉树节点
struct TreeNode* createNode(int val) {
struct TreeNode* newNode = (struct TreeNode*)malloc(sizeof(struct TreeNode));
newNode->val = val;
newNode->left = NULL;
newNode->right = NULL;
return newNode;
}
// 向二叉排序树插入节点
struct TreeNode* insertNode(struct TreeNode* root, int val) {
if (root == NULL) {
return createNode(val);
}
if (val < root->val) {
root->left = insertNode(root->left, val);
} else if (val > root->val) {
root->right = insertNode(root->right, val);
}
return root;
}
// 求指定节点在二叉排序树中所在层次
int getNodeLevel(struct TreeNode* root, int val, int level) {
if (root == NULL) {
return 0; // 节点不存在,返回 0
}
if (val == root->val) {
return level; // 找到节点,返回当前层次
} else if (val < root->val) {
return getNodeLevel(root->left, val, level + 1); // 在左子树中递归查找
} else {
return getNodeLevel(root->right, val, level + 1); // 在右子树中递归查找
}
}
int main() {
// 创建二叉排序树
struct TreeNode* root = NULL;
root = insertNode(root, 4);
insertNode(root, 2);
insertNode(root, 1);
insertNode(root, 3);
insertNode(root, 6);
insertNode(root, 5);
insertNode(root, 7);
// 指定节点的值
int targetValue = 3;
// 求指定节点在二叉排序树中所在层次
int level = getNodeLevel(root, targetValue, 1);
if (level > 0) {
printf("节点 %d 在二叉排序树中的层次为 %d\n", targetValue, level);
} else {
printf("节点 %d 不存在于二叉排序树中\n", targetValue);
}
// 释放二叉树的内存
// 这里省略了释放内存的代码,实际使用中应该释放二叉树的内存
return 0;
}