给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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. 数据结构
- 包含左右指针,一个标志量(来判断这个节点是否被访问过),数据节点的Data
typedef int ElementType;
typedef struct TreeNode *Tree;
struct TreeNode{
ElementType Data;
Tree Left;
Tree Right;
int flag; //判别一个序列 没被访问,设为0,被访问,设为1
};
typedef Tree BinTree;
2. 建立搜索树
- 创建搜索树的函数,传入N,输入搜索树节点信息。输入根节点Data,便利用NewNode()函数申请内存来创建一个节点;之后输入剩余的N-1个节点信息,将N-1个节点插入到二叉搜索树中(Insert()函数)
- Insert函数:先判断根节点是否为空,如果为空,便申请一个节点空间;若不为空,则判断输入节点的Data和根节点的Data大小:
1)若T->Data < V,则应将V插入到根节点的右边,因此传入T->Right
2)若T->Data > V,则应将V插入到根节点的左边,因此传入T->Left
3.
BinTree Create(int N)
{
// 最好一开始声明为NULL
BinTree T = NULL;
int V,i;
scanf("%d",&V);
// 创建节点
T = NewNode(V);
// 将剩下的N-1个节点插入到二叉搜索树中
for(i = 0;i < N-1;i++)
{
scanf("%d",&V);
T = Insert(T,V);
}
return T;
}
/* 常规操作 申请一个struct TreeNode大小的内存,创建新结点*/
BinTree NewNode(int V)
{
BinTree BT;
BT = (BinTree)malloc(sizeof(struct TreeNode));
BT->Data = V;
BT->Left = NULL;
BT->Right = NULL;
BT->flag = 0;
return BT;
}
/*迭代终止为T为NULL时*/
BinTree Insert(BinTree T,int V)
{
BinTree TNode;
if(!T) T = NewNode(V);
else{
if(T->Data < V)
{
T->Right = Insert(T->Right,V);
}
if(T->Data > V)
{
T->Left = Insert(T->Left,V);
}
}
return T;
}
3. 判断
函数Judge :
- 读入第一个数据,判断是否和根节点Data相同,若不相同,则说明两个二叉树不一致!!(flag设为1),若相同,则继续往下走
- 看看剩下的N-1个节点一致吗。BT->flag:若是访问过此节点,则设为1。
- 方法便是看看输入的节点Data在原始序列中路径中,是否出现了flag=0的情况,若出现,则说明不一致。
下图为Judge函数步骤:
代码
/*当发现序列中的某个数和T不一样,必须把序列后面的数都读完*/
int Judge(BinTree T,int N)
{
int i;
int V;
int flag = 0;
/*flag:0 代表目前一致 flag:1 代表已经不一致 */
scanf("%d",&V);
if(V!=T->Data) flag = 1;
else T->flag = 1;
for (i = 1; i < N;i++)
{
scanf("%d",&V);
/* 若check为0,则return 0;之前出现的节点没有被访问过,出现新的节点,所以树不一致*/
if(!flag && !check(T,V)) flag = 1;
}
if(flag)
return 0;
else return 1;
}
int check(BinTree T,int V)
{
if(T->flag)
{
if(V > T->Data) return check(T->Right,V);
if(V < T->Data) return check(T->Left,V);
// 若相等,则说明之前已经出现过一个相同的数,则两个树不一致
if(V == T->Data) return 0;
}
else{
if(V == T->Data)
{
T->flag = 1;
return 1;
}
else{
return 0;
}
}
}
4. 完整代码
#include <stdio.h>
#include <stdlib.h>
typedef int ElementType;
typedef struct TreeNode *Tree;
struct TreeNode{
ElementType Data;
Tree Left;
Tree Right;
int flag; //判别一个序列 没被访问,设为0,被访问,设为1
};
typedef Tree BinTree;
BinTree Create(int N);
BinTree NewNode(int V);
BinTree Insert(BinTree T,int V);
int Judge(BinTree T,int N);
int check(BinTree T,int V);
void ResetT(BinTree T);
void FreeTree(BinTree T);
int main()
{
int N,L,i;
BinTree T;
scanf("%d",&N);
while(N)
{
scanf("%d",&L);
T = Create(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);
}
}
void FreeTree(BinTree T)
{
if(T->Left) FreeTree(T->Left);
if(T->Right) FreeTree(T->Right);
free(T);
}
void ResetT(BinTree T)
{
if(T->Left) ResetT(T->Left);
if(T->Right) ResetT(T->Right);
T->flag = 0;
}
BinTree Create(int N)
{
BinTree T = NULL;
int V,i;
scanf("%d",&V);
T = NewNode(V);
for(i = 0;i < N-1;i++)
{
scanf("%d",&V);
T = Insert(T,V);
}
return T;
}
BinTree NewNode(int V)
{
BinTree BT;
BT = (BinTree)malloc(sizeof(struct TreeNode));
BT->Data = V;
BT->Left = NULL;
BT->Right = NULL;
BT->flag = 0;
return BT;
}
BinTree Insert(BinTree T,int V)
{
BinTree TNode;
if(!T) T = NewNode(V);
else{
if(T->Data < V)
{
T->Right = Insert(T->Right,V);
}
if(T->Data > V)
{
T->Left = Insert(T->Left,V);
}
}
return T;
}
/*当发现序列中的某个数和T不一样,必须把序列后面的数都读完*/
int Judge(BinTree T,int N)
{
int i;
int V;
int flag = 0;
/*flag:0 代表目前一致 flag:1 代表已经不一致 */
scanf("%d",&V);
if(V!=T->Data) flag = 1;
else T->flag = 1;
for (i = 1; i < N;i++)
{
scanf("%d",&V);
/* 若check为0,则return 0;之前出现的节点没有被访问过,出现新的节点,所以树不一致*/
if(!flag && !check(T,V)) flag = 1;
}
if(flag)
return 0;
else return 1;
}
int check(BinTree T,int V)
{
if(T->flag)
{
if(V > T->Data) return check(T->Right,V);
if(V < T->Data) return check(T->Left,V);
// 若相等,则说明之前已经出现过一个相同的数,则两个树不一致
if(V == T->Data) return 0;
}
else{
if(V == T->Data)
{
T->flag = 1;
return 1;
}
else{
return 0;
}
}
}