PAT 1066

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
struct Node
{
	Node* lchild;
	Node* rchild;
	int data;
	int h;
};

vector<int> in;
int gh(Node* root)
{
	if (root == NULL)
		return 0;
	else
		return root->h;
}

int gb(Node* root)
{
	if (root == NULL)
		return 0;
	else
		return gh(root->lchild) - gh(root->rchild);
}

void uh(Node* root)
{
	if (root == NULL)
		return;
	else
		root->h = max(gh(root->lchild),gh(root->rchild))+1;
}

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

void R(Node* &root)
{
	Node* temp;
	temp = root->lchild;
	root->lchild = temp->rchild;
	temp->rchild = root;
	uh(root);
	uh(temp);
	root = temp;
	return;
}

void insert(Node* &root,int a)
{
	if (root == NULL)
	{
		root = new Node;
		root->lchild = root->rchild = NULL;
		root->h = 1;
		root->data = a;
		return;
	}
		if (a < root->data)
		{
			insert(root->lchild, a);
			uh(root);
			if (gb(root) == 2)
			{
				if (gb(root->lchild) == 1)
					R(root);
				else if (gb(root->lchild) == -1)
				{
					L(root->lchild);
					R(root);
				}
			}	
		}
		else
		{
			insert(root->rchild, a);
			uh(root);
			if (gb(root) == -2)
			{
				if (gb(root->rchild) == -1)
					L(root);
				else if (gb(root->rchild) == 1)
				{
					R(root->rchild);
					L(root);
				}
			}
		}
}

int main()
{
	int n;
	cin >> n;
	int temp;
	for (int i=0;i<n;i++)
	{
		cin >> temp;
		in.push_back(temp);
	}
	Node* root = NULL;
	for (int i=0;i<in.size();i++)
	{
		insert(root,in[i]);
	}
	cout << root->data;
	system("pause");
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值