算法笔记-问题 B: 二叉搜索树

问题 B: 二叉搜索树

题目描述

判断两序列是否为同一二叉搜索树序列

输入

开始一个数n，(1<=n<=20) 表示有n个需要判断，n= 0 的时候输入结束。
接下去一行是一个序列，序列长度小于10，包含(0~9)的数字，没有重复数字，根据这个序列可以构造出一颗二叉搜索树。
接下去的n行有n个序列，每个序列格式跟第一个序列一样，请判断这两个序列是否能组成同一颗二叉搜索树。

输出

如果序列相同则输出YES，否则输出NO

样例输入 Copy

6
45021
12045
54120
45021
45012
21054
50412
0

样例输出 Copy

NO
NO
YES
NO
NO
NO

代码：

#include <bits/stdc++.h>

using namespace std;

struct node{
    int data;
    node *lchild, *rchild;
};

/*插入*/
void insert(node* &root, int data){
    if(root == NULL){
        root = new node;
        root->data = data;
        root->lchild = root->rchild = NULL;
        return;
    }
    if(data == root->data) return;/*关键！！这里默认二叉排序树不允许有重复键值*/
    if(data<root->data) insert(root->lchild, data);
    else insert(root->rchild, data);
}

/*先序*/
void preOrder(node* root, vector<int> &vi){
    if(root == NULL) return;
    vi.push_back(root->data);
    preOrder(root->lchild, vi);
    preOrder(root->rchild, vi);
}

int main(){
    int n;
    while(scanf("%d", &n)!=EOF){
        if(n==0) break;
        node* root = NULL;
        vector<int> origin;
        string s;
        cin >> s;
        for(int i=0; i<s.length(); i++){
            insert(root, s[i]-'0');
        }
        preOrder(root, origin);
        for(int i=0; i<n; i++){
            node* newroot = NULL;
            vector<int> pre;
            string str;
            cin>>str;
            for(int i=0; i<s.length(); i++){
                insert(newroot, str[i]-'0');
            }
            preOrder(newroot, pre);
            if(pre == origin)
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
	return 0;
}