将给定的一系列数字插入初始为空的AVL树,请你输出最后生成的AVL树的根结点的值。
输入格式:
输入的第一行给出一个正整数N(≤20),随后一行给出N个不同的整数,其间以空格分隔。
输出格式:
在一行中输出顺序插入上述整数到一棵初始为空的AVL树后,该树的根结点的值。
输入样例1:
5
88 70 61 96 120
输出样例1:
70
输入样例2:
7
88 70 61 96 120 90 65
输出样例2:
88
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<string>
using namespace std;
typedef struct node* AVLNode;
struct node{
AVLNode Left,Right;
int Data;
};
int GetHeight(AVLNode A){
if(!A) return 0;
else{
int L=GetHeight(A->Left);
int R=GetHeight(A->Right);
return L>R?(L+1):(R+1);
}
}
AVLNode SingleLeftRotation(AVLNode A){
AVLNode B=A->Left; //左单旋,顺时针,B是A的左,A的左是B的右,B的右是A
A->Left=B->Right;
B->Right=A;
return B;
}
AVLNode SingleRightRotation(AVLNode A){
AVLNode B=A->Right;//右单旋,B是A的右,A的右是B的左,B的左是A
A->Right=B->Left;
B->Left=A;
return B;
}
AVLNode DoubleLeftRightRotation(AVLNode A){
A->Left=SingleRightRotation(A->Left); //左右双旋,A的左go右单旋,返回A的左单旋
return SingleLeftRotation(A);
}
AVLNode DoubleRightLeftRotation(AVLNode A){
A->Right=SingleLeftRotation(A->Right); //右左双旋,A的右go左单旋,返回A的右单旋
return SingleRightRotation(A);
}
AVLNode Insert(AVLNode A,int x){
if(!A){ //A->malloc!!!
AVLNode A=(AVLNode)malloc(sizeof(struct node));
A->Data=x;
A->Left=A->Right=NULL;
return A;
}
else if(x<A->Data){
A->Left=Insert(A->Left,x);
if(GetHeight(A->Left)-GetHeight(A->Right)==2){
if(x<A->Left->Data) A=SingleLeftRotation(A);
else A=DoubleLeftRightRotation(A);
}
}
else if(x>A->Data){
A->Right=Insert(A->Right,x);
if(GetHeight(A->Left)-GetHeight(A->Right)==-2){
if(x>A->Right->Data) A=SingleRightRotation(A);
else A=DoubleRightLeftRotation(A);
}
}
return A;
}
int main(){
int n,x;
AVLNode A=NULL; //!!!
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&x);
A=Insert(A,x);
}
printf("%d",A->Data);
return 0;
}
总结:为了避免野指针,定义指针的时候必须给指针初始化(以防指针空间的数据没有及时清空),用free(P)释放掉指针所指的内存空间后,必须立即同时把p赋值为NULL,避免后面程序指针P而导致意想不到的错误,甚至系统崩溃!