原题如下:
判断两序列是否为同一二叉搜索树序列
INPUT:
开始一个数n,(1<=n<=20) 表示有n个需要判断,n= 0 的时候输入结束。
接下去一行是一个序列,序列长度小于10,包含(0~9)的数字,没有重复数字,根据这个序列可以构造出一颗二叉搜索树。
接下去的n行有n个序列,每个序列格式跟第一个序列一样,请判断这两个序列是否能组成同一颗二叉搜索树。
OUTPUT:
如果序列相同则输出YES,否则输出NO
Sample Input:
2
567432
543267
576342
2
567432
543267
576342
0
Sample Output:
YES
NO
YES
NO
这道题的题解不少。但一直有一个争议:到底应该是用两个序比较还是用一个序比较?
真相只有一个:若你构造的是二叉搜索树,那么仅需要一次前序比较即可;若你是随机构造的一棵二叉树,那么至少需要前序和中序两次比较(这里的前序也可以换成中序)
今天我构造了一棵二叉搜索树,然后仅仅通过前序比较AC了此题
二叉搜索树的介绍和构造方法可以参考我的上一篇博客。
二叉搜索树的介绍和构造方法可以参考我的上一篇博客。
二叉搜索树的介绍和构造方法可以参考我的上一篇博客。
代码
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>
using namespace std;
#define NIL 0x3f
int start=1;
void TREE_INSERT(int*KEY,int*LEFT,int*RIGHT,int N)
{
for(int i=2;i<=N;i++)
{
int y=NIL,x=1;
while(x!=NIL )
{
y=x;
if(KEY[i]<KEY[x]) x=LEFT[x];
else x=RIGHT[x];
}
if (KEY[i]<KEY[y]) LEFT[y]=i;//树的根为KEY[1]是既定事实,所以i直接从2开始与KEY[1]开始比较,也不可能存在y==NIL的情况,NIL设置大了后在这里爆栈
else RIGHT[y]=i;
}
}
void OUT_PUT(int*KEY,int*LEFT,int*RIGHT,int*SAVE,int k)//我试着用DFS方法输出,但是每次多输出一个0
{
SAVE[start++]=KEY[k];
if(LEFT[k]!=NIL) OUT_PUT(KEY,LEFT,RIGHT,SAVE,LEFT[k]);
if(RIGHT[k]!=NIL) OUT_PUT(KEY,LEFT,RIGHT,SAVE,RIGHT[k]);
}
void INPUT(char*import,int N,int*KEY,int*LEFT,int*RIGHT)
{
start=1;
fill(LEFT,LEFT+(N+2),NIL);
fill(RIGHT,RIGHT+(N+2),NIL);
scanf("%s",import);
for(int i=1;i<=N;i++) KEY[i]=import[i-1]-48;
}
int main()
{
int M,N=0;
char import[1050];
while(scanf("%d",&M) && M!=0)
{
scanf("%s",import);
N=strlen(import);
int*KEY=(int*)malloc(sizeof(int)*(N+2));
for(int i=1;i<=N;i++) KEY[i]=import[i-1]-48;//初始化KEY数组
int *LEFT=(int*)malloc(sizeof(int)*(N+2));fill(LEFT,LEFT+(N+2),NIL);//见鬼呀哈哈,原来memset不是填值呀,填值应当用fill
int *RIGHT=(int*)malloc(sizeof(int)*(N+2));fill(RIGHT,RIGHT+(N+2),NIL);
int *SAVE=(int*)malloc(sizeof(int)*(2*N+2));fill(SAVE,SAVE+(2*N+2),NIL);//SAVE存储第一列的后序遍历
int *MMP=(int*)malloc(sizeof(int)*(2*N+2));fill(MMP,MMP+(2*N+2),NIL);//MMP存储要与第一列比较的后序遍历
TREE_INSERT(KEY,LEFT,RIGHT,N);
OUT_PUT(KEY,LEFT,RIGHT,SAVE,1);
for(int i=1;i<=M;i++)
{
int grd=0;
INPUT(import,N,KEY,LEFT,RIGHT);
TREE_INSERT(KEY,LEFT,RIGHT,N);
OUT_PUT(KEY,LEFT,RIGHT,MMP,1);
for(int j=1;j<=N;j++)//这次只比较前序看看
{
if(SAVE[j]!=MMP[j]) {printf("NO\n");grd++;break;}
}
if(!grd) printf("YES\n");
}
free(KEY);free(LEFT);free(RIGHT);free(SAVE);free(MMP);start=1;//这里一定要把该还原的都还原,否则下一组数据会出错
}
return 0;
}
还是这句话:有问题欢迎留言!