【PTA】是否同一棵二叉搜索树-不建树思路

/**********************************************************************************************************************
给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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
**********************************************************************************************************************/
//以第一个数为根结点,所有比根结点大的数作为右子树序列,否则为左子树序列
//“抽取”所有右子树结点,得到:哨兵->右子树,此时原序列剩下:根结点->左子树
//对右子树、左子树递归排序,续接到根结点,释放哨兵
//最终得到序列的先序遍历
//以先序遍历的序列判断是否同一棵搜索树(先序遍历确定搜索树)

#include <iostream>
using namespace std;

typedef struct TNode* Tree;
typedef struct TNode {
    
    
	int Data;
	Tree Next;
};

Tree InputOrder(int N)
{
    
    
	int X;
	Tree root, tmp, input;
	root = new struct TNode({
    
    0, NULL});
	tmp = root;
	for (int n(0); n < N; n++)
	{
    
    
		cin >> X;
		input = new struct TNode({
    
     X, NULL });
		tmp->Next = input;
		tmp = input;
	}
	tmp = root;
	root = root->Next;
	delete tmp;
	return root;
}

Tree PreOrder(Tree T)
{
    
    
	//以第一个数为根结点,所有比根结点大的数作为右子树序列,否则为左子树序列
	//“抽取”所有右子树结点,得到:哨兵->右子树,此时原序列剩下:根结点->左子树
	//对右子树、左子树递归排序,续接到根结点,释放哨兵
	//最终得到序列的先序遍历
	Tree Root(T);
	if (T) {
    
    
		Tree Right = new struct TNode({
    
     T->Data, NULL }), RightRoot(Right);
		while(T->Next)
		{
    
    
			if (T->Next->Data > Root->Data){
    
    
				Right->Next = T->Next;
				T->Next = T->Next->Next;
				Right = Right->Next;
			}
			else
				T = T->Next;
		}
		Right->Next = NULL;	//尾部指向NULL
		Right = RightRoot->Next;	//“抽取”右子树
		delete RightRoot;	//释放哨兵

		Tree Left(Root->Next);
		Root->Next = PreOrder(Left);	//对左子树递归排序,根结点指向左子树
		for (Left = Root; Left->Next; Left = Left->Next);	//移动到左子树尾部
		Left->Next = PreOrder(Right);	//对右子树递归排序,左子树尾部指向右子树
	}
	return Root;
}

bool Compare(Tree T1, Tree T2)
{
    
    
	while (T1 && T2 && T1->Data == T2->Data)
	{
    
    
		T1 = T1->Next;
		T2 = T2->Next;
	}
	return !(T1 || T2);	//若T1, T2均为空,则所有结点均相等
}

int main()
{
    
    
	int N, L;

	cin >> N;
	while (N) {
    
    
		cin >> L;
		Tree RefOrder = InputOrder(N);
		RefOrder = PreOrder(RefOrder);
		while (L--) {
    
    
			Tree CmpOrder = InputOrder(N);
			CmpOrder = PreOrder(CmpOrder);
			if (Compare(RefOrder, CmpOrder))
				cout << "Yes\n";
			else
				cout << "No\n";
		}
		cin >> N;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/jimaofu0494/article/details/103058214