7-4 是否同一棵二叉搜索树 (25分)

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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>
typedef struct TreeeNode *Tree;
struct TreeeNode{
	Tree Left,right;
	int v;
	int flag;
};
Tree insert(Tree T,int v)
{
	if(!T)
	{
		Tree t=(Tree)malloc(sizeof(struct TreeeNode));
		t->v=v;
		t->flag=0;
		t->Left=t->right=NULL;
		return t;
	}
	else
	{
		if(v>T->v)
		{
			T->right=insert(T->right,v);
		}
		else if(v<T->v)
		{
			T->Left=insert(T->Left,v);
		}
		return T;
	}
}
Tree create(int N)
{
	Tree T=(Tree)malloc(sizeof(struct TreeeNode));
	int v;scanf("%d",&v);
	T->v=v;
	T->flag=0;
	T->Left=T->right=NULL;
	int i;
	for(i=1;i<N;i++)
	{
		scanf("%d",&v);
		insert(T,v);
	}
	return T;
}
int check(Tree T,int v)
{
	if(!T)return 0;
	if(T->v==v)
	{
		if(!T->flag)
		{
			T->flag=1;
			return 1;
		}
		else
		{
			return 0;
		}
	}
	else
	{
		if(!T->flag)return 0;
		else
		{
			if(T->v<v)
			{	
				return check(T->right,v);
			}
			else
			{
				return check(T->Left,v);
			}
		}
	}
}
int judge(Tree T,int N)
{
	int i,v;
	int flag=1;
	for(i=0;i<N;i++)
	{
		scanf("%d",&v);
		int f=check(T,v);
		if(!f)
		{
			flag=0;
			break;
		}
	}
	if(flag)
	{
		return 1;
	}
	else
	{
		int j;
		for(j=0;j<N-i-1;j++)
		{
			scanf("%d",&v);
		}
		return 0;
	}
}
void reset(Tree T)
{
	if(!T)return;
	else
	{
		T->flag=0;
		reset(T->Left);
		reset(T->right);
	}
}
void FreeTree(Tree T)
{
	if(!T)return;
	else
	{
		FreeTree(T->Left);
		FreeTree(T->right);
		free(T);
	}
}
int main()
{
	int N,L;
	while(1)
	{
		scanf("%d%d",&N,&L);
		if(N==0)break;
		/*先读入待比较序列,建树*/
		Tree T=create(N);
		/*读入比较序列,判断*/
		int i;
		for(i=0;i<L;i++)
		{
			int f=judge(T,N);
			if(f)
			{
				printf("Yes\n");
			}
			else 
			{
				printf("No\n");
			}
			reset(T);
		}
		FreeTree(T);
	}
	
	return 0;
}
发布了97 篇原创文章 · 获赞 12 · 访问量 2429

猜你喜欢

转载自blog.csdn.net/weixin_43301333/article/details/103981939