题目
来源
3540. 二叉搜索树 - AcWing题库
思路
建立二叉搜索树(注意传参时用到了引用,可以直接对root进行修改),同时进行递归遍历;遍历可以分前中后三种写,也可以用标志来代替合在一起。其余详见代码。
代码
#include<bits/stdc++.h>
using namespace std;
const int N=110;
int l[N],r[N],w[N],idx;//l存储左子树,r存储右子树,w存储节点上的值
int root=0;
void insert(int& u,int x){ //加个引用,直接在root上可以做修改
if(u==0) u=++idx,w[u]=x; //根节点为空,那么就需要建树
else if(x<w[u])insert(l[u],x);
else if(x>w[u])insert(r[u],x);
//如果相等不需要处理
}
void dfs(int u,int t){
if(u==0)return;
if(t==0)cout<<w[u]<<" ";//前序遍历,先输出根节点
dfs(l[u],t);//遍历左子树
if(t==1)cout<<w[u]<<" ";//中序遍历,输出根节点
dfs(r[u],t);
if(t==2)cout<<w[u]<<" ";
}
int main(){
int n;
cin>>n;
while(n--){
int x;
cin>>x;
insert(root,x);
}
for(int i=0;i<3;i++)//做出三种遍历操作
{
dfs(root,i); //i=0,1,2分别表示前序,中序,后序遍历
cout<<endl;
}
return 0;
}