数据结构实验之查找一:二叉排序树
题目描述
对应给定的一个序列可以唯一确定一棵二叉排序树。然而,一棵给定的二叉排序树却可以由多种不同的序列得到。例如分别按照序列{3,1,4}和{3,4,1}插入初始为空的二叉排序树,都得到一样的结果。你的任务书对于输入的各种序列,判断它们是否能生成一样的二叉排序树。
输入
输入包含若干组测试数据。每组数据的第1行给出两个正整数N (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 <cstdio> #include <algorithm> #include <iostream> #define lson l , m , rt << 1 #define rson m + 1 , r , rt << 1 | 1 #define LL long long struct node{ int data; node *l,*r; }*root_1,*root_2; int arr[20]; void insert_node(struct node *root, int data){ node *curr = NULL; node *tmp = NULL; if (root == NULL){ return; } curr = new node; curr->data = data; tmp = root; while (tmp != NULL){ if (tmp->data > curr->data && tmp->l == NULL){ tmp->l = curr; return; } else if(tmp->data <= curr->data){ while (tmp != NULL) { if (tmp->data <= curr->data && tmp->r == NULL){ tmp->r = curr; return; } else if (tmp->data > curr->data){ break; } else { tmp = tmp->r; } } } else { tmp = tmp->l; } } } struct node *build(const int *arr, int len) { struct node *root = NULL; int i = 0; if (len <= 0) { return NULL; } root = new node; root->data = arr[0]; for (i = 1; i < len; i++) { insert_node(root, arr[i]); } return root; } bool Judge(node *root1,node *root2){ if(root1 == NULL && root2 == NULL){ return true; } else if(root1 == NULL || root2 == NULL) return false; else{ if(root1->data != root2->data) return false; bool is_left = Judge(root1->l,root2->l); bool is_right = Judge(root1->r,root2->r); if(is_left && is_right) return true; else{ is_right = Judge(root1->r,root2->l); is_left = Judge(root1->l,root2->r); if (is_left&&is_right) return true; else return false; } } } int main() { int n,m; while(scanf("%d",&n) && n){ scanf("%d",&m); for(int i=0;i<n;i++){ scanf("%d",&arr[i]); } root_1 = build(arr,n); while(m--){ for(int i=0,j;i<n;i++){ scanf("%d",&arr[i]); } root_2 = build(arr,n); if(Judge(root_1,root_2)){ printf("Yes\n"); } else{ printf("No\n"); } } } return 0; }