PTA_2019春_044_是否同一棵二叉搜索树

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而，一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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 treenode {
	int data;
	int left;
	int right;
}Tree;
void issametree(int N,int L);/*判断是否同树*/
void insert(int root, Tree* tree, int X);/*生成二搜索树的插入操作*/ 
int main() {
	int N; /*n个节点*/
	int L; /*L个待测序列*/


	for (int k = 0;; k++) {
		scanf("%d", &N);
		if (N == 0)  break;
		scanf("%d", &L);
		issametree(N, L);
	}

	return 0;
}
void issametree(int N,int L) {
	/*先将节点存在数组里*/
	int* array = (int*)malloc(N * sizeof(int));
	for (int i = 0; i < N; i++) {
		scanf("%d", &array[i]);
	} 

	/*先将所给节点构成二叉搜索树*/
	Tree* tree = (struct treenode*)malloc((N+1) * sizeof(struct treenode));
	for (int i = 1; i < N + 1; i++) {
		tree[i].data = i;
		tree[i].left = -1;
		tree[i].right = -1;        /*一开始测试不行，原来是我把这一行的i，写成1，我是智障*/ 
	}  /*初始化树*/

	int root = array[0]; /*根节点*/
	for (int i = 1; i < N; i++) {
		insert(root, tree, array[i]);
	}/*原树建立完成*/
	
	/*下面则完成L个需要判断是否同树的序列 即tree 与 tree1*/

	for (int i = 0; i < L; i++) {
		/*先将所给节点构成二叉搜索树*/
		Tree* tree1 = (struct treenode*)malloc((N + 1) * sizeof(struct treenode));
		for (int j = 1; j< N + 1; j++) {
			tree1[j].data = i;
			tree1[j].left = -1;
			tree1[j].right = -1;
		}  /*初始化树*/
		int* array1 = (int*)malloc(N * sizeof(int));
		for (int j = 0; j < N; j++) {
			scanf("%d", &array1[j]);
		}
		int root1 = array1[0]; /*根节点*/
		for (int j = 1; j < N; j++) {
			insert(root1, tree1, array1[j]);
		}/*树tree1建立完成*/

		/*开始比较*/
		
		for (int j = 1; j < N + 1; j++) {
			if ((tree[j].left == tree1[j].left) && (tree[j].right) == tree1[j].right) {
				if (j == N) {
					printf("Yes\n");  /*我是智障，题中输出为Yes，我却强行YES，怎么能跑对！！！*/
				}
			}
			else {          
				printf("No\n");
				break;                /*此处原来没有加break，导致NO输出过多*/
			}
		}
	}


}
void insert(int root, Tree* tree, int X) {
	while (root != -1) {
		if (X < root) {
			if (tree[root].left == -1) {/*没有左子树，则可以插入*/
				tree[root].left = X;
				break;
			}
			else root = tree[root].left;
		}
		else { /*X>root*/
			if (tree[root].right == -1) { /*没有右子树，则插入*/
				tree[root].right = X;
				break;
			}
			else root = tree[root].right;
		}
	}
}
/*
4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0
*/