04-树4 是否同一棵二叉搜索树

04-树4 是否同一棵二叉搜索树

来自：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

程序代码：

/*
	题目：对于不同的输入序列，判断是否为同一个二叉搜索树。
	求解思路：
			1、搜素树表示；
			2、建搜索树T；
			3、判断一序列是否与搜索树T一致。
*/

#include <stdio.h>
#include <stdlib.h>

//搜索树表示
typedef struct TreeNode *Tree;
struct TreeNode
{
	int v;
	Tree Left, Right;
	int flag;
};
Tree MakeTree(int N);
Tree NewNode(int V);
Tree Insert(Tree T, int V);
int Judge(Tree T, int N);
void ResetT(Tree T);
void FreeTree(Tree T);

//程序框架搭建
int main()
{
	int N, L, i;
	Tree T;

	scanf("%d", &N);
	while (N)
	{
		scanf("%d", &L);
		T = MakeTree(N);
		for (i = 0; i < L; i++)
		{
			if (Judge(T, N))
			{
				printf("Yes\n");
			}
			else
			{
				printf("No\n");
			}
			ResetT(T); //清除T中的标记flag
		}
		FreeTree(T);
		scanf("%d", &N);
	}
	return 0;
}

//如何建搜索数
Tree MakeTree(int N)
{
	Tree T;
	int i, V;

	scanf("%d", &V);
	T = NewNode(V);
	for (i = 1; i < N; i++)
	{
		scanf("%d", &V);
		T = Insert(T, V);
	}
	return T;
}

//建新节点（一般为第一个节点）
Tree NewNode(int V)
{
	Tree T = (Tree)malloc(sizeof(struct TreeNode));
	T->v = 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->v)
		{
			T->Right = Insert(T->Right, V);
		}
		else
		{
			T->Left = Insert(T->Left, V);
		}
	}
	return T;
}

//如何判别（已经访问过的节点要被标记）
int check(Tree T, int V)
{
	if (T->flag)
	{
		if (V < T->v)
		{
			return check(T->Left, V);
		}
		else if (V > T->v)
		{
			return check(T->Right, V);
		}
		else
		{
			return 0;
		}
	}else
	{
		if (V==T->v)
		{
			T->flag = 1;
			return 1;
		}else
		{
			return 0;
		}
	}
}

/*
//如何判别（判别输入的树是否一致）
int Judge(Tree T, int N)
{
	int i, V;
	//有bug版本
	scanf("%d", &V); //当发现序列中的某个数与T不一致时，
	if (V != T->v)	 //必须把序列后边的数都读完。
	{
		return 0;
	}
	else
	{
		T->flag = 1;
	}

	for (i = 0; i < N; i++)
	{
		scanf("%d", &V);
		if (!check(T, V))
		{
			return 0;
		}
	}

	return 1;
}
*/

int Judge(Tree T, int N)
{
	int i, V, flag = 0;
	//flag:0代表目前还一致，1代表已经不一致

	scanf("%d", &V);
	if (V != T->v)
	{
		flag = 1;
	}
	else
	{
		T->flag = 1;
	}
	for (i = 1; i < N; i++)
	{
		scanf("%d", &V);
		if ((!flag) && (!check(T, V)))
		{
			flag = 1;
		}
	}
	if (flag)
	{
		return 0;
	}
	else
	{
		return 1;
	}
}

//清楚T中各个节点的标记
void ResetT(Tree T)
{
	if (T->Left)
	{
		ResetT(T->Left);
	}
	if (T->Right)
	{
		ResetT(T->Right);
	}
	T->flag = 0;
}

void FreeTree(Tree T)
{
	if (T->Left)
	{
		FreeTree(T->Left);
	}
	if (T->Right)
	{
		FreeTree(T->Right);
	}
	free(T);
}