c++ 创建二叉树函数 二叉树查找指定结点的双亲和兄弟的函数 并用主函数测试
时间: 2023-11-26 20:03:01 浏览: 87
用C++实现二叉树的查找功能
4星 · 用户满意度95%
好的,我来为您介绍如何实现这个功能。
首先,我们需要定义二叉树的结构体,包括数据域和左右子树指针:
```cpp
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
```
接下来,我们可以编写一个创建二叉树的函数,可以通过输入数组来构建二叉树:
```cpp
TreeNode* buildTree(vector<int>& nums, int index) {
if (index >= nums.size() || nums[index] == -1) {
return NULL;
}
TreeNode* root = new TreeNode(nums[index]);
root->left = buildTree(nums, 2 * index + 1);
root->right = buildTree(nums, 2 * index + 2);
return root;
}
```
其中,-1 表示空节点。
接下来,我们可以编写一个查找指定结点的双亲和兄弟的函数:
```cpp
void findParentAndSibling(TreeNode* root, int target, TreeNode* parent, TreeNode* &sibling) {
if (root == NULL || root->val == target) {
return;
}
if (root->left != NULL && root->left->val == target) {
parent = root;
sibling = root->right;
return;
}
if (root->right != NULL && root->right->val == target) {
parent = root;
sibling = root->left;
return;
}
findParentAndSibling(root->left, target, root, sibling);
findParentAndSibling(root->right, target, root, sibling);
}
```
其中,parent 和 sibling 都是引用类型,用于记录找到的双亲和兄弟结点。
最后,我们可以在主函数中测试这两个函数:
```cpp
int main() {
vector<int> nums = {1, 2, 3, -1, -1, 4, 5};
TreeNode* root = buildTree(nums, 0);
TreeNode* parent = NULL;
TreeNode* sibling = NULL;
findParentAndSibling(root, 4, parent, sibling);
if (parent == NULL) {
cout << "4 不在树中" << endl;
} else {
cout << "4 的双亲为 " << parent->val << endl;
if (sibling == NULL) {
cout << "4 没有兄弟" << endl;
} else {
cout << "4 的兄弟为 " << sibling->val << endl;
}
}
return 0;
}
```
输出结果为:
```
4 的双亲为 2
4 的兄弟为 5
```
这样,我们就成功实现了创建二叉树的函数和查找指定结点的双亲和兄弟的函数,并用主函数测试了它们的功能。
阅读全文