平衡二叉树的基本操作

1、结点数据结构定义

struct Node
{
	int v;
	int height;
	Node* lchild;
	Node* rchild;
};

2、新建结点

Node* newNode(int v)
{
	Node* node=new Node;
	node->v=v;
	node->height=1;
	node->lchild=node->rchild=NULL;
	return node;
}

3、获取结点高度

int getHeight(Node* root)
{
	if(root==NULL) return 0;
	return root->height;
}

4、获取平衡因子

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

5、更新高度

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

6、左旋

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

7、插入结点

void insert(Node* &root,int v) 
{
	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)
			{
				R(root);
			}
			else if(getBalanceFactor(root->lchild)==-1)
			{
				L(root->lchild);
				R(root);
			}
		}		
	}
	else
	{
		insert(root->rchild,v);
		updateHeight(root);
		if(getBalanceFactor(root)==2)
		{
			if(getBalanceFactor(root->rchild)==-1)
			{
				L(root);
			}
			else if(getBalanceFactor(root->rchild)==1)
			{
				R(root->rchild);
				L(root);
			}
		}
	}	
}