查看原题
题意
先给你一个数字n,接着是一串数字,你把它顺序扫描后建立一棵二叉搜索树。然后再陆续给你n串数字,判断它们尽力的二叉树和第一个二叉树是否相同。
题目保证给出的字符串长度不大于10.每个数字在0~9之间。且给出的数字不会有重复。
分析
很明显,这是一个数据结构的题目,很容易想到的就是直接自定义个二叉树结点的结构来做了。但是我感觉那样好麻烦,题目已经给出了数字长度不会大于10.所以直接用静态的链表也就是数组来解就可以了。要是数字的个数太多就不能这样做了。来看二叉树中结点与数组下标的对应关系。
注意!上图中结点中的数字并不代表结点的值,而是代表与保存结点值的数组的下标之间的对应关系。
- 假设用于存储的数组为node。那么根节点保存在node[0],而左儿子保存在node[1],右儿子保存在node[2]。
- 可知父节点与子节点之间的换算关系是这样的:若父节点为 i ,那么左儿子为(i+1)*2-1,右儿子为(i+1)*2。
- 所以我们要做的就是分别用两个int数组来保存前后两个字符串建立的二叉树。然后去比较这两个数组是否相等就可以了。
- 那么我们的node数组需要多大呢?10?9?错错错。这里要考虑所有可能的结点插入情况。其中最极端一种就是有9个结点,但是全部是右子树,这是最浪费空间的情况。此时树的深度是9。考虑深度为9的满二叉树的结点个数:2^9-1。只要大于这个数字就好了。
代码
#include <iostream>
using namespace std;
#include <cstring>
#define MAX 512
int node[MAX];
int node2[MAX];
void makeTree(char *str,int *node)
{
for(int i=0;str[i]!='\0';i++)
{
int j=0;
int t=str[i]-'0';
while(node[j]!=-1)
{
if(t>node[j])
j=(j+1)*2;
if(t<node[j])
j=(j+1)*2-1;
}
node[j]=t;
}
}
void compare()
{
for(int i=0;i<MAX;i++)
if(node[i]!=node2[i])
{
cout<<"NO"<<endl;
return;
}
cout<<"YES"<<endl;
}
int main()
{
int n;
char str[12],str2[12];
cin>>n;
if(!n)
return 0;
memset(node,-1,sizeof(node));
cin>>str;
makeTree(str,node);
int len=strlen(str);
while(n--)
{
memset(node2,-1,sizeof(node2));
cin>>str2;
int len2=strlen(str2);
if(len!=len2)
{
cout<<"NO"<<endl;
continue;
}
makeTree(str2,node2);
compare();
}
main();
return 0;
}
PS:
这里因为n有多个,一般的main函数章写法就是while(cin>>n){ }了,但是我不想套这么大的一个花括号,于是就写了个递归调用main()。哈哈O(∩_∩)O~