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

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);
}