数据结构实验之查找一:二叉排序树 C
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 <stdio.h>
#include <stdlib.h>
struct node
{
int data,d;
struct node *l,*r;
};
int deep(struct node *root)
{
if(root==NULL)
return -1;
else
return root->d;
}
struct node *LL(struct node *root)
{
struct node *p;
p=root->l;
root->l=p->r;
p->r=root;
if(deep(root->l)>deep(root->r))
root->d=deep(root->l)+1;
else
root->d=deep(root->r)+1;
if(deep(p->l)>deep(p->r))
p->d=deep(p->l)+1;
else
p->d=deep(p->r)+1;
return p;
}
struct node *RR(struct node *root)
{
struct node *p;
p=root->r;
root->r=p->l;
p->l=root;
if(deep(root->l)>deep(root->r))
root->d=deep(root->l)+1;
else
root->d=deep(root->r)+1;
if(deep(p->l)>deep(p->r))
p->d=deep(p->l)+1;
else
p->d=deep(p->r)+1;
return p;
}
struct node *LR(struct node *root)
{
root->l=RR(root->l);
return LL(root);
}
struct node *RL(struct node *root)
{
root->r=LL(root->r);
return RR(root);
}
struct node *creat(struct node *root,int x)
{
if(root==NULL)
{
root=(struct node *)malloc(sizeof(struct node));
root->d = 0; //会忘
root->data=x;
root->l=root->r=NULL;
}
else if(x<root->data)
{
root->l=creat(root->l,x);
if(deep(root->l)-deep(root->r)==2)
{
if(x<root->l->data)
root=LL(root);
else
root=LR(root);
}
}
else if(x>root->data)
{
root->r=creat(root->r,x);
if(deep(root->r)-deep(root->l)==2)
{
if(x>root->r->data)
root=RR(root);
else
root=RL(root);
}
}
if(deep(root->l)>deep(root->r))
root->d=deep(root->l)+1;
else
root->d=deep(root->r)+1;
return root;
};
int main()
{
int n,i,x;
struct node *root;
root=NULL;
scanf("%d",&n);
for(i=0; i<n; i++)
{
scanf("%d",&x);
root=creat(root,x);
}
printf("%d",root->data);
return 0;
}
/***************************************************
User name: cb020502
Result: Accepted
Take time: 0ms
Take Memory: 144KB
Submit time: 2018-08-15 10:23:30
****************************************************/