题目描述
输入一系列整数,建立二叉排序树,并进行前序,中序,后序遍历。
输入描述:
输入第一行包括一个整数n(1<=n<=100)。
接下来的一行包括n个整数。
输出描述:
可能有多组测试数据,对于每组数据,将题目所给数据建立一个二叉排序树,并对二叉排序树进行前序、中序和后序遍历。
每种遍历结果输出一行。每行最后一个数据之后有一个空格。
输入中可能有重复元素,但是输出的二叉树遍历序列中重复元素不用输出。
示例1
输入
5
1 6 5 9 8
输出
1 6 5 9 8
1 5 6 8 9
5 8 9 6 1
思路:
递归遍历树,循环插入树节点
代码:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <cstdio>
using namespace std;
struct node{
int id;
node *left, *right;
node(int i):id(i),left(NULL),right(NULL){}
};
node* insert_BST(int id, node *root){
if(root == NULL){
root = new node(id);
return root;
}
node* temp = root;
node* parent;
//记录插入节点是插在父节点的右边还是左边
//约定flag为0的时候在父节点的左边插入新节点,flag为1的时候在父节点的右边插入新节点
int flag = 0;
while(temp != NULL){
parent = temp;
if(id > temp->id){
flag = 1;
temp = temp->right;
}else if(id < temp->id){
flag = 0;
temp = temp->left;
}else{ //树中不可有重复元素
return root;
}
}
if(flag == 1){
parent->right= new node(id);
}else{
parent->left = new node(id);
}
return root;
}
void preOrder(node *root)
{
if (root == NULL)
{
return;
}
printf("%d ", root->id);
preOrder(root->left);
preOrder(root->right);
}
void inOrder(node *root)
{
if (root == NULL)
{
return;
}
inOrder(root->left);
printf("%d ", root->id);
inOrder(root->right);
}
void postOrder(node *root)
{
if (root == NULL)
{
return;
}
postOrder(root->left);
postOrder(root->right);
printf("%d ", root->id);
}
int main(){
int n;
int a[105];
while(cin>>n){
node *root = NULL;
for(int i = 0; i < n; i++){
int x;
cin>>x;
a[i] = x;
}
for(int i = 0; i < n; i++){
root = insert_BST(a[i], root);
}
preOrder(root);
printf("\n");
inOrder(root);
printf("\n");
postOrder(root);
printf("\n");
}
return 0;
}