浙江大学数据结构MOOC-课后习题-第四讲-树4 是否同一棵二叉搜索树

题目汇总
浙江大学数据结构MOOC-课后习题-拼题A-代码分享-2024

题目描述

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而，一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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>
#include <vector>
#define MAXSIZE 10
struct treeNode
{
	int value;
	treeNode *left, * right;
	int flag;
};
typedef treeNode* tree;

void freeTree(tree T)
{
	if (T)
	{
		freeTree(T->left);
		freeTree(T->right);
		free(T);
	}
}
void resetTree(tree T)
{
	if (T)
	{
		resetTree(T->left);
		resetTree(T->right);
		T->flag = 0;
	}
}
tree newNode(int V)
{
	tree t = (tree)malloc(sizeof(treeNode));
	t->value = V;
	t->left = t->right = NULL;
	t->flag = 0;
	return t;
}

tree insert(tree T, int V)
{
	if (!T) T = newNode(V);
	else
	{
		if (V > T->value)
			T->right = insert(T->right, V);
		else
			T->left = insert(T->left, V);
	}
	return T;
}
tree buildTree(int N)
{
	//建立一个头结点
	int V;
	tree T;
	std::cin >> V;
	T = newNode(V);
	//插入新节点
	for (int i = 1; i < N; i++)
	{
		std::cin >> V;
		T = insert(T, V);
	}
	//返回根节点
	return T;
}

bool check(tree T, int V)
{
	if (T->flag)
	{
		if (V > T->value)
			return check(T->right, V);
		else if (V < T->value)
			return check(T->left, V);
		else return false;
	}
	else
	{
		if (V == T->value)
		{
			T->flag = 1;
			return true;
		}
		return false;
	}
}


bool judge(tree T, int N)
{
	int V;
	int flag = 0; //0表示目前是一致的 1表示是不一致的
	std::cin >> V;
	if (V != T->value) flag = 1;
	else
	{	
		T->flag = 1;
	}
	for (int i = 1; i < N; i++)
	{
		std::cin >> V;
		if (!flag && !check(T, V))
		{
			flag = 1;
		}
	}
	if (flag) return false;
	else return true;

}

int main()
{	
	int N, L;
	tree T;
	std::vector<int> res;
	//读取命令行输入
	std::cin >> N;
	while (N)
	{	
		std::cin >> L;
		T = buildTree(N);//建立作为基准的二叉搜索树
		//判定当前序列是否符合要求
		for (int i = 0; i < L; i++)
		{
			if (judge(T, N)) res.push_back(1);
			else res.push_back(0);
			resetTree(T);
		}
		freeTree(T);
		std::cin >> N;
	}
	//打印结果
	for (size_t i = 0; i < res.size(); i++)
	{
		if (res[i] == 1)
			std::cout << "Yes" << std::endl;
		else
			std::cout << "No" << std::endl;
	}

}