04-树4 是否同一棵二叉搜索树
分数 25
作者 陈越
单位 浙江大学
给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。
输入格式:
输入包含若干组测试数据。每组数据的第1行给出两个正整数N (≤10)和L,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出N个以空格分隔的正整数,作为初始插入序列。随后L行,每行给出N个插入的元素,属于L个需要检查的序列。
简单起见,我们保证每个插入序列都是1到N的一个排列。当读到N为0时,标志输入结束,这组数据不要处理。
输出格式:
对每一组需要检查的序列,如果其生成的二叉搜索树跟对应的初始序列生成的一样,输出“Yes”,否则输出“No”。
输入样例:
4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0
输出样例:
Yes
No
No
鸣谢青岛大学周强老师补充测试数据!
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
C++ (g++)
思路:
创建两棵树,一棵给出的树,另一棵是那被检查的L个树。
创建树BST,为最初始的那个树,然后输入L个树,如果匹配的话,输出yes,否则输出no。最后释放这个新的树T。
比较完L个树后,释放树BST,进行新的N的比较。
创建树:
树初始化为空,读入数据,这里我们用到插入函数。
插入节点:
如果树为空,直接申请空间建立新的树。
如否则小于插入左子树 ,大于插入右子树,最后返回根结点。
释放树:
释放左树,释放右树。
检查是否为同一棵二叉搜索树:
如果都为空,肯定是。
如果一个空一个不空,肯定不是。
如果根结点不相等,肯定不是。
否则继续比较树的左节点和树的右节点。
AC代码:
#include<bits/stdc++.h>
using namespace std;
typedef struct TreeNode *BinTree;
typedef BinTree Position;
struct TreeNode{
int Data;
BinTree left;
BinTree right;
};
BinTree CreateBST(int N);
BinTree Insert(int x,BinTree BST);
void FreeTree(BinTree T);
int check(BinTree BST,BinTree T);
int main()
{
int N,L,i;
cin>>N;
BinTree BST,T;
while(N){
cin>>L;
BST=CreateBST(N);
for(i=0;i<L;i++){
T=CreateBST(N);
if(check(BST,T)) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
FreeTree(T);
}
FreeTree(BST);
cin>>N;
}
return 0;
}
BinTree CreateBST(int N){
//每一次循环读取一个数,逐个插入搜索树
int i,x;
BinTree BST=NULL;
for(i=0;i<N;i++){
//一次循环读取一行N个数
cin>>x;
BST=Insert(x,BST);
}
return BST;
}
BinTree Insert(int x,BinTree BST){
//将读取到的数插入树
if(!BST){//指针为空,直接申请节点并将数存入Data
BST=(BinTree)malloc(sizeof(struct TreeNode));
BST->Data=x;
BST->left=NULL;
BST->right=NULL;
}else if(x<BST->Data)//小于插入左子树
BST->left=Insert(x,BST->left);
else if(x>BST->Data)//大于插入右子树
BST->right=Insert(x,BST->right);
return BST;
}
void FreeTree(BinTree T){//每次比较结束后需要把后树释放
if(T->left) FreeTree(T->left);
if(T->right) FreeTree(T->right);
free(T);
}
int check(BinTree BST,BinTree T){
//比较两树,所有节点位置数量,data大小
if((!BST&&T)||(BST&&!T)) return 0;
if(!BST&&!T) return 1;
if(BST->Data!=T->Data) return 0;
return check(BST->left,T->left)&&check(BST->right,T->right);
}