DS二叉排序树之查找
题目描述
给出一个数据序列,建立二叉排序树,并实现查找功能
输入
第1行输入n,表示首个序列包含n个数据
第2行输入n个数据,都是自然数且互不相同,数据之间用空格隔开
第3行输入m,表示要查找m个数据
接着输入m行,每行一个要查找的数据,都是自然数
以此类推输入下一个示例
输出
第一行输出有序的数据序列,对二叉排序树进行中序遍历可以得到
从第二行起,输出查找结果,如果查找成功输出查找次数,如果查找失败输出-1
输入样例1:
6
22 33 55 66 11 44
7
11
22
33
44
55
66
77
输出样例1:
11 22 33 44 55 66
2
1
2
4
3
4
-1
输入样例2:
6
33 22 55 11 66 44
4
88
11
44
66
输出样例2:
11 22 33 44 55 66
-1
3
3
3
参考代码:
#include <iostream>
#include <iomanip>
#include <vector>
using namespace std;
int times;
int sizes;
struct Node{
int data;
Node *left;
Node *right;
};
Node *createNode(int data){
Node *node = new Node;
node->data = data;
node->left = nullptr;
node->right = nullptr;
return node;
}
void insertNode(Node *root,int data){
if(data < root->data){
if(root->left == nullptr)
root->left = createNode(data);
else
insertNode(root->left,data);
} else {
if(root->right == nullptr)
root->right = createNode(data);
else
insertNode(root->right,data);
}
}
void preOrder(Node *root){
if(root == nullptr)
return;
preOrder(root->left);
cout << root->data << " ";
preOrder(root->right);
}
void find(Node *root,int data){
if (data < root->data){
times++;
if (root->left == nullptr){
times = -1;
return;
}
find(root->left,data);
} else if (data > root->data){
times++;
if (root->right == nullptr){
times = -1;
return;
}
find(root->right,data);
} else if (data == root->data){
times++;
return;
}
}
int main(){
int n;
cin >> n;
sizes = n;
vector<int> nums(n);
for(int i = 0;i < n;i++){
cin >> nums[i];
}
Node *root = createNode(nums[0]);
for(int i = 1;i < n;i++){
insertNode(root,nums[i]);
}
preOrder(root);
cout << endl;
int m;
cin >> m;
while (m--){
int num;
cin >> num;
times = 0;
find(root,num);
cout << times << endl;
}
return 0;
}