PAT Root of AVL Tree(C++实现)

最新推荐文章于 2025-06-02 13:01:50 发布

Moonlightbear

最新推荐文章于 2025-06-02 13:01:50 发布

阅读量193

点赞数

文章标签： c++ 算法数据结构

本文链接：https://blog.youkuaiyun.com/Moonlightbear/article/details/120657851

版权

该博客介绍了AVL树的基本概念，包括左单旋、右单旋、左右双旋和右左双旋四种旋转操作，以及它们在插入元素时保持树平衡的作用。代码展示了AVL树节点的定义、树高的计算、插入操作以及各种旋转函数的实现，最后通过一个示例展示了如何插入元素并输出根节点数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接7-6 Root of AVL Tree (25 分)

这道题本质上是在考察平衡二叉树的调整：左单旋、右单旋、左右双旋、右左双旋。

运行结果

在这里插入图片描述

我的代码

可改进项： 左单旋、右单旋、左右双旋、右左双旋函数可改为私有成员函数；
返回根节点的数据可考虑写为函数，毕竟直接返回引用在事实上打破了类的封装性。

#include<iostream>
using namespace std;

//返回较大值
int Max(int a, int b)
{
	return a > b ? a : b;
}

//结点
struct Node {
	int Data;
	Node * Left;
	Node * Right;
	int Height;
};
using Position = Node * ;


class AVLTree
{
public:
	AVLTree();
	~AVLTree();
	Position& GetRoot();
	const int GetHeight(const Position& TreeNode);
	void Insert(Position& Root, const int X);
	//左单旋
	Position SingleLeftRotation(Position A);
	//右单旋
	Position SingleRightRotation(Position A);
	//左右双旋
	Position DoubleLeftRightRotation(Position A);
	//右左双旋
	Position DoubleRightLeftRotation(Position A);
	//销毁
	void Destroy(Position& Root);
private:
	Position Root;
};

AVLTree::AVLTree() : Root { nullptr } {}

AVLTree::~AVLTree()
{
	Destroy(Root);
}

//返回根结点的引用
Position& AVLTree::GetRoot() { return Root; }

//求以某个结点为根节点的树高
const int AVLTree::GetHeight(const Position& TreeNode)
{
	if (!TreeNode)
		return 0;
	return Max(GetHeight(TreeNode->Left), GetHeight(TreeNode->Right)) + 1;
}


void AVLTree::Insert(Position& Root, const int X)
{
	if (!Root) {
		Root = new Node;
		Root->Data = X;
		Root->Height = 1;
		Root->Left = Root->Right = nullptr;
	}
	else if (X < Root->Data) {
		Insert(Root->Left, X);
		if (GetHeight(Root->Left) - GetHeight(Root->Right) == 2)
			if (X < Root->Left->Data)
				//左单旋
				Root = SingleLeftRotation(Root);
			else
				//左右双旋
				Root = DoubleLeftRightRotation(Root);
	}
	else if (X > Root->Data) {
		Insert(Root->Right, X);
		if (GetHeight(Root->Right) - GetHeight(Root->Left) == 2)
			if (X > Root->Right->Data)
				//右单旋
				Root = SingleRightRotation(Root);
			else
				//右左双旋
				Root = DoubleRightLeftRotation(Root);
	}
	Root->Height = Max(GetHeight(Root->Left), GetHeight(Root->Right)) + 1;
}

//左单旋
Position AVLTree::SingleLeftRotation(Position A)
{
	Position B = A->Left;
	A->Left = B->Right;
	B->Right = A;
	A->Height = Max(GetHeight(A->Left), GetHeight(A->Right)) + 1;
	B->Height = Max(GetHeight(B->Left), A->Height) + 1;
	return B;
}

//右单旋
Position AVLTree::SingleRightRotation(Position A)
{
	Position B = A->Right;
	A->Right = B->Left;
	B->Left = A;
	A->Height = Max(GetHeight(A->Left), GetHeight(A->Right)) + 1;
	B->Height = Max(GetHeight(B->Right), A->Height) + 1;
	return B;
}

//左右双旋
Position AVLTree::DoubleLeftRightRotation(Position A)
{
	A->Left = SingleRightRotation(A->Left);
	return SingleLeftRotation(A);
}

//右左双旋
Position AVLTree::DoubleRightLeftRotation(Position A)
{
	A->Right = SingleLeftRotation(A->Right);
	return SingleRightRotation(A);
}

void AVLTree::Destroy(Position& Root)
{
	if (!Root)
		return;
	Destroy(Root->Left);
	Destroy(Root->Right);
	delete Root;
	Root = nullptr;
}


int main()
{
	int N, X;
	cin >> N;
	AVLTree T{};
	for (int i = 0; i < N; ++i) {
		cin >> X;
		T.Insert(T.GetRoot(), X);
	}
	cout << T.GetRoot()->Data;
}