|AVL模板|A 1066 Root of AVL Tree

最新推荐文章于 2023-08-15 14:14:45 发布

PDYJ

最新推荐文章于 2023-08-15 14:14:45 发布

阅读量96

点赞数

分类专栏：算法笔记数据结构 PAT甲级

本文链接：https://blog.csdn.net/weixin_44769957/article/details/109271433

版权

算法笔记同时被 3 个专栏收录

78 篇文章 7 订阅

订阅专栏

数据结构

64 篇文章 1 订阅

订阅专栏

PAT甲级

31 篇文章 0 订阅

订阅专栏

link

//注意传入的是 指向node指针的引用
//因为要修改节点
void L(node* &root){
    node* tmp=root->rchild;
    root->rchild=tmp->lchild;
    tmp->lchild=root;
    updateHeight(root);
    updateHeight(tmp);
    root=tmp;
}
//注意传入的是 指向node指针的引用
//因为要修改节点
void R(node* &root){
    node* tmp=root->lchild;
    root->lchild=tmp->rchild;
    tmp->rchild=root;
    updateHeight(root);
    updateHeight(tmp);
    root=tmp;
}

int getHeight(node* root){
    if(root==NULL)return 0;
    else return root->h;
}

void updateHeight(node* root){
    root->h=max(getHeight(root->lchild),getHeight(root->rchild))+1;
}

int getBalanceFactor(node* root){
    return getHeight(root->lchild)-getHeight(root->rchild);
}


void insert(node* &root,int v){
    if(root==NULL){
        root=new node;
        root->v=v;
        root->h=1;
        root->lchild=root->rchild=NULL;
        return;
    }
  if(v<root->v){
       insert(root->lchild,v);
       updateHeight(root);//2
    if(getBalanceFactor(root)==2){//1-0=1
        if(getBalanceFactor(root->lchild)==1){
            //LL
            R(root);
        }else if(getBalanceFactor(root->lchild)==-1){
            //LR
            L(root->lchild);
            R(root);
        }
    } 
   }else{
       insert(root->rchild,v);
       updateHeight(root);
    if(getBalanceFactor(root)==-2){
        if(getBalanceFactor(root->rchild)==-1){
            //RR
            L(root);
        }else if(getBalanceFactor(root->rchild)==1){
            //RL
            R(root->rchild);
            L(root);
        }    
    }
  }   
}



#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n;

struct node{
    int v,h;
    node* lchild;
    node* rchild;
};

int getHeight(node* root){
    if(root==NULL)return 0;
    else return root->h;
}

void updateHeight(node* root){
    root->h=max(getHeight(root->lchild),getHeight(root->rchild))+1;
}

int getBalanceFactor(node* root){
    return getHeight(root->lchild)-getHeight(root->rchild);
}

void L(node* &root){
    node* tmp=root->rchild;
    root->rchild=tmp->lchild;
    tmp->lchild=root;
    updateHeight(root);
    updateHeight(tmp);
    root=tmp;
}

void R(node* &root){
    node* tmp=root->lchild;
    root->lchild=tmp->rchild;
    tmp->rchild=root;
    updateHeight(root);
    updateHeight(tmp);
    root=tmp;
}

void insert(node* &root,int v){
    if(root==NULL){
        root=new node;
        root->v=v;
        root->h=1;
        root->lchild=root->rchild=NULL;
        return;
    }
    
   if(v<root->v){
       insert(root->lchild,v);
       updateHeight(root);
    if(getBalanceFactor(root)==2){
        if(getBalanceFactor(root->lchild)==1){
            //LL
            R(root);
        }else if(getBalanceFactor(root->lchild)==-1){
            //LR
            L(root->lchild);
            R(root);
        }
    } 
   }else{
       insert(root->rchild,v);
       updateHeight(root);
    if(getBalanceFactor(root)==-2){
        if(getBalanceFactor(root->rchild)==-1){
            //RR
            L(root);
        }else if(getBalanceFactor(root->rchild)==1){
            //RL
            R(root->rchild);
            L(root);
        }    
    }
  }
    
    
    
}


int main(){
    cin>>n;
    node* root=NULL;
    for(int i=0;i<n;i++){
        int temp;
        cin>>temp;
        insert(root,temp);
    }
    cout<<root->v;
    return 0;
}


------------
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
struct node
{
	int v;//结点数据 
	int height;//结点高度 
	node* lchild;
	node* rchild;
} *root;

node* newnode(int v)//创建一个新结点，新结点的高度置为1 
{
	node* Node = new node;
	Node->v = v;
	Node->height = 1;
	Node->lchild = Node->rchild = NULL;
	return Node;
}

int getHeight(node* root)//获得当前结点高度 
{
	if (root == NULL)
		return 0;
	return root->height;
}

void updateHeight(node* root)//更新当前根节点高度 
{
	root->height = max(getHeight(root->lchild), getHeight(root->rchild)) + 1;
}

int getBalanceFactor(node* root)//获取平衡因子，通过左子树-右子树高度 
{
	return getHeight(root->lchild) - getHeight(root->rchild);
}
//左旋，temp指向根节点的右子树，共有三个步骤 
//1.让temp的左子树成为root的右子树
//2.让root成为temp的左子树
//3.将temp设置为新的root 
void L(node*& root)//采用node* &root是为了使root作为一个指针进行操作 
{
	node* temp = root->rchild;
	root->rchild = temp->lchild;
	temp->lchild = root;
	updateHeight(root);
	updateHeight(temp);
	root = temp;
}

void R(node*& root)
{
	node* temp = root->lchild;
	root->lchild = temp->rchild;
	temp->rchild = root;
	updateHeight(root);
	updateHeight(temp);
	root = temp;
}
//左旋和右旋是互逆操作，把所有的left换成right，right换成left就行了 
void insert(node*& root, int v)//按BST规则插入，并进行适当操作使得该树始终是AVL树 
{
	if (root == NULL)
	{
		root = newnode(v);
		return;
	}
	if (v < root->v)
	{
		insert(root->lchild, v);
		updateHeight(root);//注意不要忘记更新结点高度 
		if (getBalanceFactor(root) == 2)
		{
			if (getBalanceFactor(root->lchild) == 1)//LL
			{
				R(root);
			}
			else if (getBalanceFactor(root->lchild) == -1)//LR
			{
				L(root->lchild);
				R(root);
			}
		}
	}
	else
	{
		insert(root->rchild, v);
		updateHeight(root);
		if (getBalanceFactor(root) == -2)
		{
			if (getBalanceFactor(root->rchild) == -1)//RR
			{
				L(root);
			}
			else if (getBalanceFactor(root->rchild) == 1)//RL
			{
				R(root->rchild);//RL情况经过一次右旋会变为RR，然后左旋即可 
				L(root);
			}
		}
	}
}
int main()
{
	int n, data;
	cin >> n;
	for (int i=0; i < n; i++)
	{
		cin >> data;
		insert(root, data);
	}
	cout << root->v << endl;
	return 0;
}

左右:
1.左旋-》左左
2.（左左）右旋
右左：
1.右旋-》右右
2.（右右）左旋

由于在构建平衡二叉树的时候，
当有新节点插入时，都会判断插入后时候平衡，
这说明了插入新节点前，都是平衡的，也即高度差绝对值不会超过1。

当新节点插入后，有可能会有导致树不平衡，
这时候就需要进行调整，而可能出现的情况就有4种，
分别称作左左，左右，右左，右右。

左左即为在原来平衡的二叉树上，
在节点的左子树的左子树下，有新节点插入，
导致节点的左右子树的高度差为2，
如上即为”10“节点的左子树”7“，的左子树”4“，
插入了节点”5“或”3“导致失衡。

左左调整其实比较简单，
只需要对节点进行右旋即可，如下图，对节点”10“进行右旋，
注意：如果对左右旋变换还不是很懂或不是很熟练的，
可以对照着前面的那两个动图去想象，自己动手变换几次，就明白了。

LL：对root进行右旋
在这里插入图片描述左右：

LR：先对root->lchild进行左旋,再对root进行右旋

在这里插入图片描述即先对上图的节点”7“进行左旋，使得二叉树变成了左左，
之后再对”11“节点进行右旋，此时二叉树就调整完成，如下图，

右左
RL：先对root->rchild进行右旋，再对root进行左旋

右左即为在原来平衡的二叉树上，
在节点的右子树的左子树下，有新节点插入，
导致节点的左右子树的高度差为2，
如上即为”11“节点的右子树”15“，的左子树”13“，
插入了节点”12“或”14“导致失衡。

前面也说了，右左跟左右其实互为镜像，
所以调整过程就反过来，先对节点”15“进行右旋，
使得二叉树变成右右，之后再对”11“节点进行左旋，
此时二叉树就调整完成，如下图，

在这里插入图片描述

右右：
RR：对root进行左旋
在这里插入图片描述

右右即为在原来平衡的二叉树上，
在节点的右子树的右子树下，有新节点插入，
导致节点的左右子树的高度差为2，如上即为”11“节点的右子树”13“，
的左子树”15“，插入了节点”14“或”19“导致失衡。

右右只需对节点进行一次左旋即可调整平衡，如下图，对”11“节点进行左旋。

在这里插入图片描述

PDYJ

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
|AVL模板|A 1066 Root of AVL Tree

link#include<iostream>#include<stdio.h>#include<stdlib.h>#include<algorithm>using namespace std;struct node{ int v;//结点数据 int height;//结点高度 node* lchild; node* rchild;} *root;node* newnode(int v)//创建一个新结点，新结点的高度置为1
复制链接

扫一扫

专栏目录