HDU 3791 比较两棵二叉搜索树是否相同(建树,比较)

二叉搜索树

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 2538    Accepted Submission(s): 1102

Problem Description

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

Input

开始一个数n，(1<=n<=20) 表示有n个需要判断，n= 0 的时候输入结束。

接下去一行是一个序列，序列长度小于10，包含(0~9)的数字，没有重复数字，根据这个序列可以构造出一颗二叉搜索树。

接下去的n行有n个序列，每个序列格式跟第一个序列一样，请判断这两个序列是否能组成同一颗二叉搜索树。

Output

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

Sample Input

2
567432
543267
576342
0

Sample Output

YES
NO

Source

浙大计算机研究生复试上机考试-2010年

慕课上的视频教程

对于比较BST是否相同,有以下三种方法

1.分别建两棵搜索树进行比较

2.不建树的方法

比如2143 和2413 

可以看出2是根节点,然后第一组比2小的为1,比2大的为43; 第二组也是,所以这两棵二叉树相同.

3.建一棵树,再判别其他序列是否与该树一致(会在每个节点用到一个flag进行标记)

思路:

这道题的数据量很小,所以可以直接采用第一种方法

(下面用三种方法实现这个问题)

下面是第一种方法

//hdu 3791 两棵二叉搜索树的比较
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
using namespace std;
//http://blog.csdn.net/linraise/article/details/19573361
//这道题不知道二叉树的长度,所以应该用字符串储存
struct BST{
char key;
BST *left;
BST *right;
BST(char x):key(x),left(NULL),right(NULL){}
//	BST(){}
//	BST(int k){
//		key=x;left=right=NULL;
//	}
};

void insert (BST *&root1,char key)
{
if (key<root1->key)
{
if (root1->left)
insert(root1->left,key);
else
root1->left=new BST(key);
}
else
{
if (root1->right)
insert(root1->right,key);
else
root1->right=new BST(key);
}
}

void Del(BST* root)//这段是可以不要的
{
if (root->left) Del(root->left);
if (root->right) Del(root->right);
delete root;
//	if (root)
//	{
//		Del(root->left);
//		Del(root->right);
//		delete root;
//	}
}

bool isequal(BST* root1,BST* root2)
{
if (root1&&root2)
{
if (root1->key!=root2->key)
return false;
return (isequal(root1->left,root2->left))&&(isequal(root1->right,root2->right));
}
else if (!root1 && !root2)
return true;
else
return false;
}

int main()
{
int n;
string line;
while (~scanf("%d%*c",&n),n)
{
getline(cin,line);
BST *root1=new BST(line[0]);
for (int i=1;i<line.size();i++)
insert(root1,line[i]);
BST *root2=NULL;
while (n--)
{
getline(cin,line);
root2=new BST(line[0]);
for (int i=1;i<line.size();i++)
insert(root2,line[i]);
if (isequal(root1,root2))
puts("YES");
else
puts("NO");
}
}
return 0;
}

下面是第二种方法

//hdu 3791 两棵二叉搜索树的比较
//(不用建树的方法)
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
string line;
string line1,line2;
string line3,line4;
//因为string 还不是很熟悉,不知道怎么传递参数...所以只能暂时设为全局变量这样用了...
void insert1()
{
for (int i=1;i<line.size();i++)
{
if (line[i]<line[0])
line1.push_back(line[i]);
else
line2.push_back(line[i]);
}
/*
string::iterator it=line.begin();
char x=line.begin();
for (it++;it!=line.end();it++)
{
if (*it<x)
str1.push_back(it);
else
str2.push_back(it);
}
*/
}

void insert()
{
for (int i=1;i<line.size();i++)
{
if (line[i]<line[0])
line3.push_back(line[i]);
else
line4.push_back(line[i]);
}
}

bool isequal(string str1,string str2)
{
if (str1==str2) return true;
else return false;
}
int main()
{
int n;

while (~scanf("%d%*c",&n),n)
{
getline(cin,line);//输入一行字符
line1.clear();
line2.clear();
insert1();
while (n--)
{
getline(cin,line);
line3.clear();
line4.clear();
insert();
if ((isequal(line1,line3))&&(isequal(line2,line4))) puts("YES");
else puts("NO");
}
}
return 0;
}

下面是解法三

//hdu 3791 两棵二叉搜索树的比较
//(只建一棵树)
//建树+遍历
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
struct BST{
char key;
BST *left;
BST *right;
int flag;
BST(char x):key(x),left(NULL),right(NULL),flag(0){}
};
void insert(BST *&root,char key)
{
if (key<root->key)
{
if (root->left)
insert(root->left,key);
else
root->left=new BST(key);
}
else
{
if (root->right)
insert(root->right,key);
else
root->right=new BST(key);
}
}
void reset(BST *&root)
{
if (root->left) reset(root->left);
if (root->right) reset(root->right);
root->flag=0;
}
bool isequal(BST *root,char key)
{
if (root->flag)
{
if (key<root->key)
return isequal(root->left,key);
else if (key>root->key)
return isequal(root->right,key);
else return false;
}
else
{
if (root->key==key)
{
root->flag=1;
return true;
}
else return false;
}
}
int main()
{
int n;
string line;
while (~scanf("%d%*c",&n),n)
{
getline(cin,line);
BST *root=new BST(line[0]);
for (int i=1;i<line.size();i++)
insert(root,line[i]);
while (n--)
{
reset(root);//一开始要节点的flag都置0
bool flag=true;
getline(cin,line);
for (int i=0;i<line.size();i++)
{
if (!isequal(root,line[i]))
{
puts("NO");
flag=false;
break;
}
}
if (flag==true)	puts("YES");
}
}
return 0;
}