04-树4 是否同一棵二叉搜索树

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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 <bits/stdc++.h>
using namespace std;

typedef struct Node *LNode;
struct Node{
    int data;
    LNode left,right;
    int flag;
}; 

LNode New(int data){
    LNode T;
    T=(LNode)malloc(sizeof(struct Node));
    T->data=data;
    T->left=NULL;
    T->right=NULL;
    T->flag=0;
    return T;
}
LNode Insert(LNode T,int data){
    if(T==NULL){
        T=New(data);
    }else{
        if(data>T->data){
            T->right=Insert(T->right,data);
        }else{
            T->left=Insert(T->left,data);
        }
    }
    return T;
}

LNode create(int N){
    LNode T;
    int data;
    cin>>data;
    T=New(data);
    for(int i=1;i<N;i++){
        cin>>data;
        T=Insert(T,data);
    }
    return T;
}

int check(LNode T,int data){
    if(T->flag){
        if(data<T->data){
            return check(T->left,data);
        }else if(data>T->data){
            return check(T->right,data);
        }else{
            return 0;
        }
    }else{
        if(data==T->data){
            T->flag=1;
            return 1;
        }else{
            return 0;
        }
    }
    
}

int Judge(LNode T,int N){
    int data,Seqflag=0;
    cin>>data;
    
    if(data!=T->data){
        Seqflag=1;
    }else{
        T->flag=1;
    }
    for(int i=1;i<N;i++){
        cin>>data;
        
        if(check(T,data)==0&&Seqflag==0){
            Seqflag=1;
        }
    }
    
    if(Seqflag){
        return 0;
    }else{
        return 1;
    }
}

void ResetT(LNode T){
    if(T->left){
        ResetT(T->left);
    }
    if(T->right){
        ResetT(T->right);
    }
    T->flag=0;
}
void Free(LNode T){
    if(T->left){
        Free(T->left);
    }
    if(T->right){
        Free(T->right);
    }
    free(T);
}

int main(){
    int N,L;
    LNode T;
    cin>>N;
    while(N!=0){
        cin>>L;
        T=create(N);
        
        for(int i=0;i<L;i++){
            if(Judge(T,N)){
                
                cout<<"Yes"<<endl;
            }else{
                cout<<"No"<<endl;
            }
            ResetT(T);
        }
        Free(T);
        cin>>N;
    }
    
    return 0;
}