[数据结构学习日记]是否同一棵二叉搜索树

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而，一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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 std::cin;
using std::cout;
using std::endl;
using TreeNode_ = class TreeNode;
using Tree = TreeNode_*;
class TreeNode {
public:
	int Data{ 0 };
	Tree Left{ nullptr };
	Tree Right{ nullptr };
	int flag{ 0 };//用于测试结点是否被访问过
};
Tree Insert_Tree(Tree BST, int d) {
	if (BST == nullptr) {
		BST = new TreeNode_;
		BST->Data = d;
		BST->Left = nullptr;
		BST->Right = nullptr;
	}
	else if (d < BST->Data) {
		BST->Left = Insert_Tree(BST->Left, d);
	}
	else {
		BST->Right = Insert_Tree(BST->Right, d);
	}
	return BST;
}
void Isomorphic(Tree BST, int length) {
	int d;
	int tab{ 1 };
	Tree Tmp = nullptr;
	while (length--) {
		cin >> d;
		Tmp = BST;
		while (d != Tmp->Data) {
			if (Tmp == nullptr) {
				tab = 0;
				break;
			}
			else if (d < Tmp->Data) {
				if (Tmp->flag != 1) {
					tab = 0;
					break;
				}
				else {
					Tmp->flag = 1;
					Tmp = Tmp->Left;
				}
			}
			else if (d > Tmp->Data) {
				if (Tmp->flag != 1) {
					tab = 0;
					break;
				}
				else {
					Tmp->flag = 1;
					Tmp = Tmp->Right;
				}
			}
		}
		if (Tmp) {
			Tmp->flag = 1;
		}
	}
	if (tab == 1) {
		cout << "Yes" << endl;
	}
	else {
		cout << "No" << endl;
	}
}
void Reset_Tree(Tree BST) {
	if (BST->Left && BST->Right) {
		BST->flag = 0;
		Reset_Tree(BST->Left);
		Reset_Tree(BST->Right);
	}
	else if (BST->Left) {
		BST->flag = 0;
		Reset_Tree(BST->Left);
	}
	else if (BST->Right) {
		BST->flag = 0;
		Reset_Tree(BST->Right);
	}
	else {
		return;
	}
}
int main() {
	int length, cnt, data;
	while (true) {
		cin >> length;
		if (length == 0) {
			break;
		}
		cin >> cnt;
		Tree BST = nullptr;
		for (int i = 0; i < length; i++) {
			cin >> data;
			BST = Insert_Tree(BST, data);
		}
		while (cnt--) {
			Isomorphic(BST, length);
			Reset_Tree(BST);
		}
		delete BST;
	}
	return 0;
}

通过建树的方式建立第一颗标准树，然后对接下来的序列进行判别，序列中的每个数在之前的路径当中都必须被访问过；最后记得要重置树结构当中的flag；