給定一個插入序列就可以唯一確定一棵二叉搜索樹。然而,一棵給定的二叉搜索樹卻可以由多種不同的插入序列得到。例如分別按照序列{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
鳴謝青島大學周強老師補充測試數據!
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
typedef int ElementType;
typedef struct BSTree *BST;
struct BSTree{
ElementType data;
BST left;
BST right;
};
void CheckBSTree(int N,int L);
BST CreatBSTree(int N);
void IsSameBSTree(BST Tree1,BST Tree2,bool *flag);
BST Insert(BST T,ElementType X);
int main()
{
/* 1.讀入初始序列 2.判斷它們是否能生成一樣的二叉搜索樹 3.輸出結果 */
int N,L;
scanf("%d",&N);
while(N!=0)
{
scanf("%d",&L);
CheckBSTree(N,L);
scanf("%d",&N);
}
return 0;
}
void CheckBSTree(int N,int L)
{
BST initTree=CreatBSTree(N);
BST tmpTree;
for(int j=0;j<L;j++){
tmpTree=CreatBSTree(N);
bool f= true;
IsSameBSTree(initTree,tmpTree,&f);
if(f)
printf("Yes\n");
else
printf("No\n");
}
}
BST CreatBSTree(int N)
{
BST T = NULL;
for(int i=0;i<N;i++){
ElementType X;
scanf("%d",&X);
T=Insert(T,X);
}
return T;
}
BST Insert(BST T,ElementType X)
{
if(!T){
T = (BST)malloc(sizeof(struct BSTree));
T->data = X;
T->left = T->right = NULL;
}else{
if(X>T->data){
T->right = Insert(T->right,X);
// printf("%d->right\n",T->data);
}
else if(X<T->data)
{
T->left = Insert(T->left,X);
// printf("%d->left\n",T->data);
}
}
return T;
}
void IsSameBSTree(BST Tree1,BST Tree2,bool *flag)
{
if(Tree1||Tree2){
if((Tree1==NULL&&Tree2!=NULL)||(Tree2==NULL&&Tree1!=NULL))
{
*flag = false;
// printf("NO1\n");
}
if(Tree1->data == Tree2->data){
// printf("Tree1->data = %d,Tree2->data = %d\n",Tree1->data,Tree2->data);
IsSameBSTree(Tree1->left,Tree2->left,flag);
IsSameBSTree(Tree1->right,Tree2->right,flag);
}else{
*flag = false;
// printf("NO2\n");
}
}
}