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

Posted 2020-10-23 我只有一件白T恤

7-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

鸣谢青岛大学周强老师补充测试数据！

思路：用数组pre保存每一个结点的父结点,然后在进行检查时，用数组flag标记已经输入的数组；如果某个数父亲结点还没进来它就进来了那么肯定不是同一颗树。再次鸣谢青岛大学周强老师补充测试数据！??

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
struct TNode{
    int Date;
    struct TNode *left;
    struct TNode *right;
};
typedef struct TNode *tNode;
int main()
{
    int N, L;
    int flag[20], pre[20];                //flag用来标记输入的数组，pre记录每个数的父节点

    while (cin >> N, N){
        cin >> L;
        
        memset(pre, -1, sizeof(int)* 20);
        tNode tnode = NULL;
        for (int i = 0; i < N; i++)
        {
            int n; cin >> n;
        
            if (tnode == NULL){                    //处理根结点
                tnode = new TNode;
                pre[n] = 0;
                tnode->Date = n;
                tnode->left = NULL;
                tnode->right = NULL;
                continue;
            }

            tNode temp = tnode;
            while (temp){                    //利用非递归方法建立二叉搜索树
                if (n > temp->Date){
                    if (temp->right == NULL){
                        temp->right = new TNode;
                        pre[n] = temp->Date;
                        temp->right->Date = n;
                        temp->right->left = NULL;
                        temp->right->right = NULL;
                        break;
                    }
                    else temp = temp->right;
                }
                
                if (n < temp->Date){
                    if (temp->left == NULL){
                        temp->left = new TNode;
                        pre[n] = temp->Date;
                        temp->left->Date = n;
                        temp->left->left = NULL;
                        temp->left->right = NULL;
                        break;
                    }
                    else temp = temp->left;
                }
            }
        }
        
        while (L--){
            int cnt = 1;
            memset(flag, 0, sizeof(int)* 20);
            for (int i = 0; i < N; i++)
            {
                int n; cin >> n;
                flag[n] = 1;
                if (flag[pre[n]] != 1 && pre[n] != 0)
                    cnt = 0;
            }
            if (cnt)cout << "Yes" << endl;
            else cout << "No" << endl;
        }
        delete tnode;
    }
    return 0;
}