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

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

输入格式:

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

简单起见，我们保证每个插入序列都是1到N的一个排列。当读到N为0时，标志输入结束，这组数据不要处理。

输出格式:

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

//不建树的方法判定
#include <iostream>
#include <vector>
using namespace std;

//判定两个序列插入后的二叉树是否一致
bool checked(vector<int> origin, vector<int> check);

int main()
{
    int N, L;
    cin >> N;
    while(N){
        cin >> L;
        //读入初始插入序列
        vector<int> origin(N);
        for(int i = 0; i < N; i++){
            cin >> origin[i];
        }
        //检查L个序列
        for(int i = 0; i < L; i++){
            vector<int> check(N);
            for(int j = 0; j < N; j++){
                cin >> check[j];
            }
            //判定origin和check
            if(checked(origin, check)){
                cout << "Yes" << endl;
            }else{
                cout << "No" << endl;
            }
        }
        cin >> N;
    }
    return 0;
}

//判定两个序列插入后的二叉树是否一致
bool checked(vector<int> origin, vector<int> check)
{
    int n = origin.size();
    if(n == 0) return true;
    if(n == 1) return origin[0] == check[0];
    //先比较根节点，如果不一致返回false
    if(origin[0] != check[0]) return false;
    //找到所有比根节点小的和大的分别作为左子树和右子树
    vector<int> oMin, oMax, cMin, cMax;
    
    for(int i = 1; i < n; i++){
        if(origin[i] > origin[0]) oMax.push_back(origin[i]);
        else oMin.push_back(origin[i]);
        if(check[i] > check[0]) cMax.push_back(check[i]);
        else cMin.push_back(check[i]);
    }
    return checked(oMin, cMin) && checked(oMax, cMax);
    
}