Description 用函数实现如下二叉排序树算法:
(1) 插入新结点
(2) 前序、中序、后序遍历二叉树
(3) 中序遍历的非递归算法
(4) 层次遍历二叉树
(5) 在二叉树中查找给定关键字(函数返回值为成功1,失败0)
输入格式
第一行:准备建树的结点个数n
第二行:输入n个整数,用空格分隔
第三行:输入待查找的关键字
第四行:输入待查找的关键字
第五行:输入待插入的关键字
输出格式
第一行:二叉树的先序遍历序列
第二行:二叉树的中序遍历序列
第三行:二叉树的后序遍历序列
第四行:查找结果
第五行:查找结果
第六行~第八行:插入新结点后的二叉树的先、中、序遍历序列
第九行:插入新结点后的二叉树的中序遍历序列(非递归算法)
第十行:插入新结点后的二叉树的层次遍历序列
输入样例
7
40 20 60 18 50 56 90
18
35
30
输出样例
40 20 18 60 50 56 90
18 20 40 50 56 60 90
18 20 56 50 90 60 40
1
0
40 20 18 30 60 50 56 90
18 20 30 40 50 56 60 90
18 30 20 56 50 90 60 40
18 20 30 40 50 56 60 90
40 20 60 18 30 50 90 56
思路在代码的注释中
代码如下:
#include <iostream>
#include <stack>
#include <queue>
//#include <deque>
using namespace std;
typedef struct BSTNode
{
int data;
struct BSTNode *lchild,*rchild;
}BSTNode,*BSTree;
void Pre(BSTree T)//先序遍历
{
if(T)
{
cout<<T->data<<" ";
Pre(T->lchild);
Pre(T->rchild);
}
}
void In(BSTree T)//中序遍历
{
if(T)
{
In(T->lchild);
cout<<T->data<<" ";
In(T->rchild);
}
}
void Post(BSTree T)//后序遍历
{
if(T)
{
Post(T->lchild);
Post(T->rchild);
cout<<T->data<<" ";
}
}
void Insert(BSTree &T,int e)//二叉排序树的插入操作
{
if(!T)//找到插入位置,递归结束
{
BSTree S;
S=new BSTNode; //生成新节点
S->data=e;
S->lchild=S->rchild=NULL;
T=S;//把新节点*S链接到已找到的插入位置
}
else if(e<T->data) Insert(T->lchild,e);
else if(e>T->data) Insert(T->rchild,e);
}
void CreateBSTree(BSTree &T,int n)//创建二叉排序树
{
T=NULL;
int e;
for(int i=0;i<n;i++)
{
cin>>e;
Insert(T,e);
}
}
//根据中序遍历的顺序,对于任一结点,优先访问其左孩子,而左孩子结点又可以看作一根结点
//然后继续访问其左孩子,直到遇到左孩子为空的结点才进行访问结点,
//然后按相同的规则访问其右子树
void Inf(BSTree T)//中序遍历的非递归算法
{ //对任一结点p
//若其左孩子不为空,则将p入栈并将p的左孩子置为当前的p,然后对当前的p进行相同的处理
//若其左孩子为空,取栈顶元素并作出栈操作,访问该栈顶结点,然后将当前的p置为栈顶结点的右孩子
//直到p为空且栈为空则遍历结束
stack < BSTree > s;
BSTree p;
p=T;
while(p||!s.empty())
{
while(p)
{
s.push(p);//根指针进栈
p=p->lchild;//遍历左子树
}
if(!s.empty()) //直到左孩子为空
{
p=s.top();
cout<<p->data<<" ";//访问根节点
s.pop();//退栈
p=p->rchild;//遍历右子树
}
}
}
void Floor(BSTree T)//层次遍历
{
queue <BSTree> q;
if(T) q.push(T);//根节点入队列
while(!q.empty())//队列不为空
{
cout<<q.front()->data<<" ";
if(q.front()->lchild)
q.push(q.front()->lchild);//如果左子树不为空,左孩子入队
if(q.front()->rchild)
q.push(q.front()->rchild);//如果右子树不为空,右孩子入队
q.pop();//已经遍历过的节点出队
}
}
BSTree Search(BSTree T,int e)//二叉排序树的元素查找
{
if(!T||T->data==e) return T;
else if(e< T->data) return Search(T->lchild,e);
else if(e> T->data) return Search(T->rchild,e);
}
int main()
{
int n;
cin>>n;
BSTree T;
int x,y,z;
CreateBSTree(T,n);
Pre(T);cout<<endl;
In(T);cout<<endl;
Post(T);cout<<endl;
cin>>x>>y>>z;
if(Search(T,x)) cout<<"1"<<endl;
else cout<<"0"<<endl;
if(Search(T,y)) cout<<"1"<<endl;
else cout<<"0"<<endl;
Insert(T,z);
Pre(T);cout<<endl;
In(T);cout<<endl;
Post(T);cout<<endl;
Inf(T);cout<<endl;
Floor(T);cout<<endl;
return 0;
}