04-树4 是否同一棵二叉搜索树 (25分)

来源:互联网 发布:360软件认证中心 编辑:程序博客网 时间:2024/06/05 16:00

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。
输入格式:

输入包含若干组测试数据。每组数据的第1行给出两个正整数NNN (≤10\le 10≤10)和LLL,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出NNN个以空格分隔的正整数,作为初始插入序列。最后LLL行,每行给出NNN个插入的元素,属于LLL个需要检查的序列。

简单起见,我们保证每个插入序列都是1到NNN的一个排列。当读到NNN为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 <cstdio>#include <cstdlib>typedef struct TreeNode *Tree;struct TreeNode{    int v;    Tree Left,Right;    int flag;};Tree NewNode(int V){    Tree T=(Tree)malloc(sizeof(struct TreeNode));    T->v=V;    T->Left=T->Right=NULL;    T->flag=0;    return T;}Tree Insert(Tree T,int V){  //二叉排序树的插入方式    if(!T)  T=NewNode(V);    else {        if(V>T->v)            T->Right=Insert(T->Right,V);        else T->Left=Insert(T->Left,V);    }    return T;}Tree MakeTree(int N){    Tree Root;    int i,V;    scanf("%d",&V);    Root=NewNode(V);    for(i=1;i<N;i++){        scanf("%d",&V);        Root=Insert(Root,V);    }    return Root;}int check(Tree T,int V){    if(T->flag){    //如果当前节点查找过了        if(V<T->v)  return check(T->Left,V);//此时要查找的数小于该节点的数,去左边查找        else if(V>T->v) return check(T->Right,V);//大于该节点的数,去右边查找        else return 0;//如果相等则不一致    }    else{           //如果当前节点没查找过        if(V==T->v){//如果是要查找的节点则标记为查找过            T->flag=1;            return 1;        }        else return 0;//如果当前节点没被查找过还不是要查找的节点,则不一致    }}int Judge(Tree T,int N){    int i,V,consistent=1;//    scanf("%d",&V);    if(V!=T->v) consistent=0;   //1代表目前还一致    else T->flag=1;    for(i=1;i<N;i++){        scanf("%d",&V);        if(consistent&&(!check(T,V)))   consistent=0;//如果check不一致,则整棵树不一致    }    if(consistent)  return 1;    else return 0;}void ResetT(Tree T){    if(T){        T->flag=0;        ResetT(T->Left);        ResetT(T->Right);    }}void FreeTree(Tree T){    if(T){        FreeTree(T->Left);        FreeTree(T->Right);        free(T);        T=NULL;    }}int main(){    Tree T;    int N,L;    scanf("%d",&N);    while(N){        scanf("%d",&L);        T=MakeTree(N);        for(int i=0;i<L;i++){            if(Judge(T,N))                printf("Yes\n");            else printf("No\n");            ResetT(T);      //清除flag标记        }        FreeTree(T);        //当前用例构建的树释放掉        scanf("%d",&N);    }    return 0;}
0 0