PAT | 7-4 是否同一棵二叉搜索树

题目

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而，一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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 <stdio.h>
#include <stdlib.h>

/*
1. 建二叉搜索树：根、左节点、右节点
2. 不建树：根结点、左子树、右子树
3. 建一棵树（基准），其他序列和这棵树作比较 √
*/

typedef struct TreeNode* Tree;
struct TreeNode
{
	int v;
	Tree left, right;
	int flag;
};

Tree NewNode(int v)
{
	Tree T = (Tree)malloc(sizeof(struct TreeNode));
	T->v = v;
	T->left = T->right = NULL;
	T->flag = 0; //没有搜索过，falg为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;
}

//建搜索树
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;
}

//在T中找v是否一致
int check(Tree T, int v)
{  
	if (T->flag)//当前结点访问过
	{
		if (v < T->v) //比T小，去左子树找
			return check(T->left, v);
		else if (v > T->v)//比T大，去右子树找
			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, 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);
		//现在还是一致的，但是check已经不一致了
		if ((!flag) && (!check(T, v)))
			flag = 1;
	}

	if (flag) return 0;//不一致
	else return 1;
}

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);
}

int main()
{
	/*
	1.读入n和L
	2.根据一行序列建树T
	3.判断其他的序列是不是同一棵搜索树并输出结果
	*/
	int n, l, i;
	Tree T;

	scanf("%d", &n);
	while (n) //当读到N为0时，标志输入结束，这组数据不要处理
	{
		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;
}