判断两序列是否为同一二叉搜索树序列
Input
开始一个数n,(1<=n<=20) 表示有n个需要判断,n= 0 的时候输入结束。
接下去一行是一个序列,序列长度小于10,包含(0~9)的数字,没有重复数字,根据这个序列可以构造出一颗二叉搜索树。
接下去的n行有n个序列,每个序列格式跟第一个序列一样,请判断这两个序列是否能组成同一颗二叉搜索树。
Output
如果序列相同则输出YES,否则输出NO
Sample Input
2 567432 543267 576342 0
Sample Output
YES NO
题意:
根据二叉搜索树的生成规则先将一开始的序列生成一个二叉搜索树用ss[]存储起来,
然后输入n个序列,
每一个序列都生成一个二叉搜索树建立一个临时数组存储,比较两个数组是否相同,
相同输出“YES”,不同输出“NO”。
代码:
#include<map>
#include<string>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
map<int,string>m,h;
map<string,int>w,q;
int s[15],c[15],n,l,b;
char ss[15],cc[15];
int dfs1(string t)
{
if(w.find(t)==w.end())
{
w[t]=b;
m[b]=t;
return 1;
}
if(w[t]>b)
if(dfs1(t+'0'))
return 1;
if(w[t]<b)
if(dfs1(t+'1'))
return 1;
}
int dfs2(string t)
{
if(q.find(t)==q.end())
{
q[t]=b;
h[b]=t;
return 1;
}
if(q[t]>b)
if(dfs2(t+'0'))
return 1;
if(q[t]<b)
if(dfs2(t+'1'))
return 1;
}
int main()
{
while(~scanf("%d",&n),n)
{
m.clear(); //清空
w.clear();
scanf("%s",ss); //字符串输入 ss[]
for(int i=0; ss[i]!='\0'; i++) //数字储存 s[]
s[i]=ss[i]-'0';
l=strlen(ss); //测长度
m[s[0]]="0"; // “0”是字符串常量,字符串常量是由一对双引号括起的字符序列。
w["0"]=s[0];
for(int i=1; i<l; i++)
{
b=s[i];
dfs1("0");
}
sort(s,s+l);
for(int i=0; i<n; i++)
{
scanf("%s",cc);
h.clear(); //清空
q.clear();
for(int j=0; cc[j]!='\0'; j++)
c[j]=cc[j]-'0';
h[c[0]]="0";
q["0"]=c[0];
for(int j=1; j<l; j++)
{
b=c[j];
dfs2("0");
}
int flag=0;
sort(c,c+l);
for(int j=0; j<l; j++)
if(s[j]!=c[j]||m[s[j]]!=h[c[j]])
flag=1;
if(flag)
printf("NO\n");
else
printf("YES\n");
}
}
return 0;
}