给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。
输入格式:
输入包含若干组测试数据。每组数据的第1行给出两个正整数N (≤)和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
#include<iostream>
#include<fstream>
#include<vector>
using namespace std;
struct Node{
Node* left;
Node* right;
int digit;
};
void Print(vector<int> a,int N)
{
for(int i=0;i<N;++i){
cout << a[i] << " ";
}
cout << endl;
}
void InOrder(Node* H)
{
if(H){
InOrder(H->left);
cout << H->digit << " ";
InOrder(H->right);
}
}
Node* insert(Node * Head,int digit)
{
if(Head==nullptr){
Head = new Node;
Head->left = Head->right = nullptr;
Head->digit = digit;
}else{
if(digit<Head->digit ){
Head->left = insert(Head->left ,digit);
}else if(digit>Head->digit ){
Head->right = insert(Head->right ,digit);
}
}
return Head;
}
Node* CreateBalanceTree(vector<int> v)
{
if(v.size()==0){
return nullptr;
}
Node* Head=nullptr;
for(size_t i=0;i<v.size();++i){
if(i==0){
Head = insert(nullptr,v[0]);
}else{
Head = insert(Head, v[i]);
}
}
return Head;
}
void PreOrder(Node* H,vector<int>& v,int &index)
{
if(H){
v[index++] = H->digit;
PreOrder(H->left ,v,index);
PreOrder(H->right ,v,index);
}
}
void SameBalancedBinaryTree(Node* H1,Node * H2,int N)
{
int index1 = 0, index2 = 0;
vector<int> v1(100), v2(100);
PreOrder(H1, v1, index1);
PreOrder(H2, v2, index2);
int flag = 1;
for(int i=0;i<N;++i){
if(v1[i]!=v2[i]){
flag = 0;
break;
}
}
if(flag==1){
cout << "Yes" << endl;
}else{
cout << "No" << endl;
}
}
int main()
{
//ifstream inFile("C:\\Users\\DELL\\Desktop\\in.txt",ios::in);
int N, L;
while(cin>>N>>L){
vector<int> v1;
for(int i=0;i<N;++i){
int a;
cin >> a;
v1.push_back(a);
}
Node* H1 = CreateBalanceTree(v1);
for(int j=0;j<L;++j){
vector<int> v2;
for (int k = 0; k<N; ++k) {
int a;
cin >> a;
v2.push_back(a);
}
Node* H2 = CreateBalanceTree(v2);
SameBalancedBinaryTree(H1, H2,N);
delete H2;
}
delete H1;
}
//inFile.close();
system("pause");
return 0;
}