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

谁践踏了优雅 2022-06-03 00:46 158阅读 0赞

数据结构实验之查找一：二叉排序树  
Time Limit: 400MS Memory Limit: 65536KB  
Submit Statistic  
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”。  
Example Input

4 2  
3 1 4 2  
3 4 1 2  
3 2 4 1  
2 1  
2 1  
1 2  
0  
Example Output

Yes  
No  
No  
Hint

Author

xam

#include <iostream>
    
    using namespace std;
    
    const int MAX = 15;
    typedef struct node
    {
        int d;
        struct node *r, *l;
    } tree, *ptree;
    
    ptree creat(ptree a, int x)
    {
        if(a == NULL)
        {
            a = new tree();
            a->d = x;
            a->r = a->l = NULL;
        }
        else
        {
            if(x < a->d)
            {
                a->l = creat(a->l, x);
    
            }
            else
            {
                a->r = creat(a->r, x);
            }
        }
        return a;
    }
    int judge(ptree a, ptree b)
    {
        if(a == NULL && b == NULL)
            return 1;
        else if(a != NULL && b != NULL)
        {
            if(a->d != b->d)
                return 0;
            else if(judge(a->l, b->l) && judge(a->r, b->r))
                return 1;
            else return 0;
        }
        else return 0;
    }
    int main()
    {
        int n, m, i;
        int ra[MAX], rb[MAX];
        while(cin >> n >> m && n != 0)
        {
            for(i = 0; i < n; i++)
                cin >> ra[i];
            ptree a = new tree();
            a->d = ra[0];
            a->l = a->r = NULL;
            for(i = 1; i < n; i++)
                a = creat(a, ra[i]);
            while(m--)
            {
                for(i = 0; i < n; i++)
                    cin >> rb[i];
                ptree b = new tree();
                b->d = rb[0];
                b->l = b->r = NULL;
                for(i = 1; i < n; i++)
                    b = creat(b, rb[i]);
                int k = judge(a, b);
                if(k == 1)
                cout << "Yes" << endl;
                else cout << "No" << endl;
            }
        }
        return 0;
    }