是否同一棵二叉搜索树

问题描述

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

C语言代码

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

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

/*释放t的空间*/
void freetree(tree t)
{
	if (t->left) freetree(t->left);
	if (t->right) freetree(t->right);
	free(t);
	t = NULL;
}

tree newnode(int v)
{
	tree t = (tree)malloc(sizeof(struct tree));
	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;
}

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

/*在第一棵树中搜索其他树的每个结点，若搜索所经过的每个结点在前面均出现过，则一致，否则不一致*/ 
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, 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;
}

void reset(tree t)
{
	if (t->left) reset(t->left);
	if (t->right) reset(t->right);
	t->flag = 0;
}

int main(int argc, char *argv[]) {
	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");
			reset(t);
		}
		freetree(t);
		scanf("%d", &n);
	}
	return 0;
}