数据结构实验之查找一:二叉排序树
Time Limit: 400MS
Memory Limit: 65536KB
Problem Description
对应给定的一个序列可以唯一确定一棵二叉排序树。然而,一棵给定的二叉排序树却可以由多种不同的序列得到。例如分别按照序列{3,1,4}和{3,4,1}插入初始为空的二叉排序树,都得到一样的结果。你的任务书对于输入的各种序列,判断它们是否能生成一样的二叉排序树。
Input
输入包含若干组测试数据。每组数据的第1行给出两个正整数N (n < = 10)和L,分别是输入序列的元素个数和需要比较的序列个数。第2行给出N个以空格分隔的正整数,作为初始插入序列生成一颗二叉排序树。随后L行,每行给出N个元素,属于L个需要检查的序列。
简单起见,我们保证每个插入序列都是1到N的一个排列。当读到N为0时,标志输入结束,这组数据不要处理。
Output
对每一组需要检查的序列,如果其生成的二叉排序树跟初始序列生成的二叉排序树一样,则输出"Yes",否则输出"No"。
Example Input
4 2 3 1 4 2 3 4 1 2 3 2 4 1 2 1 2 1 1 2 0
Example Output
Yes No No
#include <bits/stdc++.h> using namespace std; struct node { int data; struct node *l, *r; }; int n, m, flag; struct node *creat(struct node *root, int x) { if(!root) { root = new node; root -> data = x; root -> l = NULL; root -> r = NULL; } else { if(x > root->data) { root->r = creat(root->r, x); } else { root -> l = creat(root->l, x); } } return root; }; void judge(struct node *root, struct node *root1) { if(root != NULL && root1 != NULL) { if(root->data != root1->data) { flag =0; return ; } judge(root->l, root1->l); judge(root->r, root->r); } } int main() { int x; while(cin >> n && n != 0) { cin >> m; struct node *root; root = new node; root = NULL; for(int i = 0; i < n; i++) { cin >> x; root = creat(root, x); } while(m--) { struct node *root1; root1 = new node; root1 = NULL; flag = 1; for(int i = 0; i < n; i++) { cin >> x; root1 = creat(root1, x); } judge(root, root1); if(flag == 0) { cout << "No" << endl; } else { cout << "Yes" << endl; } } } return 0; }