数据结构实验之查找一：二叉排序树（判断两序列是否为同一棵树）

数据结构实验之查找一：二叉排序树

Problem Description

对应给定的一个序列可以唯一确定一棵二叉排序树。然而，一棵给定的二叉排序树却可以由多种不同的序列得到。例如分别按照序列{3,1,4}和{3,4,1}插入初始为空的二叉排序树，都得到一样的结果。你的任务书对于输入的各种序列，判断它们是否能生成一样的二叉排序树。

Input

输入包含若干组测试数据。每组数据的第1行给出两个正整数N (n < = 10)和L，分别是输入序列的元素个数和需要比较的序列个数。第2行给出N个以空格分隔的正整数，作为初始插入序列生成一颗二叉排序树。随后L行，每行给出N个元素，属于L个需要检查的序列。
简单起见，我们保证每个插入序列都是1到N的一个排列。当读到N为0时，标志输入结束，这组数据不要处理。

Output

对每一组需要检查的序列，如果其生成的二叉排序树跟初始序列生成的二叉排序树一样，则输出"Yes"，否则输出"No"。

Sample Input

4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0

Sample Output

Yes
No
No

根据之前对树的了解，我们知道给出一棵树的前序遍历和中序遍历，或者给出一棵树的后续遍历和中序遍历，我们能确定一棵树。

所以这道题的思路是：
对输入的数字序列构建二叉排序树，分别进行先序遍历和中序遍历（唯一构成一个二叉树）。比较两个序列的前序遍历和中序遍历是否相同，相同则是同一颗二叉搜索树。

给出代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct node {
    int date;
    struct node *l,*r;
} tree;                 //树的结构

int a1_pre[1005];   //存放树的遍历结果，方便比较。
int a1_mid[1005];
int a2_pre[1005];
int a2_mid[1005];
int cnt = 0;

tree *creat( tree*p, int key );  // 构造二叉排序树
int cmp(int a[], int b[], int len);  // 比较两个数组是否相等
void show_pre1(tree *root1);  //进行树的遍历，存到相应的数组中
void show_pre2(tree *root2);
void show_mid1(tree *root1);
void show_mid2(tree *root2);

int main()
{
    int n,l,i,key,j;
    while ( ~scanf("%d",&n) ) {
        if ( n==0 ) {
            break;
        }
        scanf("%d",&l);
        tree *root1 = NULL;
        for ( i=0; i<n; i++ ) {
            scanf("%d",&key);
            root1 = creat(root1,key);
        }
        cnt = 0;
        show_pre1(root1);
        cnt = 0;
        show_mid1(root1);

        for ( i=0; i<l; i++ ) {
            tree *root2 = NULL;
            for ( j=0; j<n; j++ ) {
                scanf("%d",&key);
                root2 = creat(root2,key);
            }
            cnt = 0;
            show_pre2(root2);
            cnt = 0;
            show_mid2(root2);

            if ( cmp(a1_pre,a2_pre,n)==1 && cmp(a1_mid,a2_mid,n) ) {
                printf("Yes\n");
            } else {
                printf("No\n");
            }
        }
    }

    return 0;
}

tree *creat( tree*p, int key )
{
    if ( p==NULL ) {
        p = (tree *)malloc(sizeof(tree));
        p -> date = key;
        p -> l = p -> r = NULL;
    } else if ( key>p->date ) {
        p -> r = creat(p->r,key);
    } else {
        p -> l = creat(p->l,key);
    }

    return p;
}

int cmp(int a[], int b[], int len)
{
    int i;
    int isp = 1;
    for ( i=0; i<len; i++ ) {
        if ( a[i]!=b[i] ) {
            isp = 0;
            break;
        }
    }

    return isp;
}

void show_pre1(tree *root1)
{
    if ( root1 ) {
        a1_pre[cnt++] = root1->date;
        show_pre1(root1->l);
        show_pre1(root1->r);
    }
}

void show_pre2(tree *root2)
{
    if ( root2 ) {
        a2_pre[cnt++] = root2->date;
        show_pre2(root2->l);
        show_pre2(root2->r);
    }
}

void show_mid1(tree *root1)
{
    if ( root1 ) {
        show_mid1(root1->l);
        a1_mid[cnt++] = root1->date;
        show_mid1(root1->r);
    }
}

void show_mid2(tree *root2)
{
    if ( root2 ) {
        show_mid2(root2->l);
        a2_mid[cnt++] = root2->date;
        show_mid2(root2->r);
    }
}


/***************************************************
User name: rj180323石旭
Result: Accepted
Take time: 0ms
Take Memory: 168KB
Submit time: 2019-01-23 09:42:29
****************************************************/