距离PAT考试还有 11天最重要的是做透每一题
(1)思路
就是考察基本的AVL树
这里主要写的是单旋转左旋和右旋
双旋转可以用其组合得到
这里要注意的是,insert,roatewithleftchild和roatewithrightchild函数都是传的引用,root初始化为0,表示插入的位置到了
所以root的值会不断改变为当前树的根
按照这个思路必须将left,right数组初始为零
顺便为了好看,可以规定height[0]为-1
#include <cstdio> #include <vector> #include <cstring> #include <algorithm> using namespace std; int n; int num=0; const int N=30; int a[N]; int left[N]; int right[N]; int height[N]; int newnode(int x) { num++; a[num]=x; left[num]=0; right[num]=0; height[num]=0; return num; } void roatewithleftchild(int& p) { int q=left[p]; left[p]=right[q]; right[q]=p; height[p]=max(height[left[p]],height[right[p]])+1; height[q]=max(height[left[q]],height[p]); p=q; } void roatewithrightchild(int& p) { int q=right[p]; right[p]=left[q]; left[q]=p; height[p]=max(height[left[p]],height[right[p]])+1; height[q]=max(height[right[q]],height[p]); p=q; } void insert(int x,int& p) { if(!p) p=newnode(x); else if(x < a[p]){ insert(x,left[p]); // if(height[left[p]] - height[right[p]] == 2) { if(x < a[left[p]]) roatewithleftchild(p); else { roatewithrightchild(left[p]); roatewithleftchild(p); } } } else if(x > a[p]) { insert(x,right[p]); if(height[right[p]] - height[left[p]] == 2) { if(x > a[right[p]]) roatewithrightchild(p); else {//left-right roate roatewithleftchild(right[p]); roatewithrightchild(p); } } } height[p]=max(height[left[p]],height[right[p]])+1; } int main() { scanf("%d",&n); memset(a,0,sizeof(a)); memset(left,0,sizeof(left)); memset(right,0,sizeof(right)); memset(height,-1,sizeof(height)); int root=0; for(int i=0;i<n;i++) { int v; scanf("%d",&v); insert(v,root); } printf("%d\n",a[root]); return 0; }
