1123 Is It a Complete AVL Tree (30分)

2020-02-02 11:02:23 阅读：249 来源： 互联网

标签：node right cur temp int 1123 Tree AVL left

在这里插入图片描述

注意事项：

插入后、旋转后要更新相应结点的高度
先更新子树高度、再更新结点高度
结点高度为该结点子树较高者高度+1

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
int N;
vector<int> traversal;
struct node {
	int val, height;
	node *left, *right;
	node(int v) {
		val = v;
		height = 1;
		left = NULL;
		right = NULL;
	}
}*root = NULL;
int getHeight(node *p) {
	return p == NULL ? 0 : p->height;
}
void updateHeight(node *&cur) {
	cur->height = max(getHeight(cur->left), getHeight(cur->right)) + 1;
}
void L(node *&p) {
	node *temp = p->right;
	p->right = temp->left;
	temp->left = p;
	p = temp;
	updateHeight(p->left);	//特别注意更新结点高度
	updateHeight(p);
}
void R(node *&p) {
	node *temp = p->left;
	p->left = temp->right;
	temp->right = p;
	p = temp;
	updateHeight(p->right);
	updateHeight(p);
}
void insert(node *&cur, int val) {
	if (cur == NULL) {
		cur = new node(val);
		return;
	}
	if (cur->val > val) {
		insert(cur->left, val);
		if (getHeight(cur->left) - getHeight(cur->right) > 1) {	//need adjustment
			if (getHeight(cur->left->left) > getHeight(cur->left->right)) {	//R
				R(cur);
			}
			else {
				L(cur->left);
				R(cur);
			}
		}
	}
	else {
		insert(cur->right, val);
		if (getHeight(cur->right) - getHeight(cur->left) > 1) {
			if (getHeight(cur->right->right) > getHeight(cur->right->left)) {
				L(cur);
			}
			else {
				R(cur->right);
				L(cur);
			}
		}
	}
	updateHeight(cur);
}
int main() {
	scanf("%d", &N);
	for (int i = 0;i < N;i++) {
		int temp;
		scanf("%d", &temp);
		insert(root, temp);
	}
	queue<node*> q;
	q.push(root);
	int level = 0;
	bool isComplete = true;
	while (!q.empty()) {
		bool isFull = true,		//标记当前层结点数是否满
			isLastLevel = true,	//标记当前层是否树的最后一层
			isEnded = false;	//对于某一层，标记各结点的子结点是否全部靠左
		int count = q.size();
		if(count != pow(2, level))
			isFull = false;
		for (int i = 0;i < count;i++) {
			node *temp = q.front();
			q.pop();
			if (temp->left != NULL || temp->right != NULL)
				isLastLevel = false;
			traversal.push_back(temp->val);
			if (temp->left != NULL) {
				if (isEnded)
					isComplete = false;
				q.push(temp->left);
			}
			else isEnded = true;
			if (temp->right != NULL) {
				if (isEnded)
					isComplete = false;
				q.push(temp->right);
			}
			else isEnded = true;
		}
		if (!isFull && !isLastLevel)
			isComplete = false;
		level++;
	}
	for (int i = 0;i < traversal.size();i++) {
		if (i != 0)
			printf(" ");
		printf("%d", traversal[i]);
	}
	if (isComplete)
		printf("\nYES\n");
	else printf("\nNO\n");
}

Ethan97 发布了77 篇原创文章 · 获赞 0 · 访问量 1320 私信关注

标签：node,right,cur,temp,int,1123,Tree,AVL,left
来源： https://blog.csdn.net/weixin_43116322/article/details/104142266

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

1123 Is It a Complete AVL Tree (30分)