1123 Is It a Complete AVL Tree (30分)

1.注意左旋右旋时，最后形成的树形式

2.层次遍历可用前序遍历存储实现，也可以用队列实现。

3.左旋即将根节点的右孩子为新的根结点，然后原root为他孩子节点的左节点，原来孩子节点的左孩子变为原root节点的右边孩子

右旋即将根节点的左孩子为新的根结点，然后原root为他孩子节点的右节点，原来孩子节点的右孩子变为原root节点的左孩子

图示理解

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>

using namespace std;
vector<int> ans;
const int N = 1<<20;
int level[N];
int n;
bool flag = true;

struct Node {
	int data;
	Node *l, *r;
	Node() {
		l = r = NULL;
	}
};
int getH(Node *root) { //计算树的高度
	if(root == NULL) return 0;
	return max(getH(root->l), getH(root->r))+1;
}
void lRotate(Node* &root) {
	Node *tmp = root->r;
	root->r = tmp->l;
	tmp->l = root;
	root = tmp;

}
void rRotate(Node* &root) {
	Node *tmp = root->l;
	root->l = tmp->r;
	tmp->r = root;
	root = tmp;
}
void lrRotate(Node* &root) {
	lRotate(root->l);
	rRotate(root);
}
void rlRotate(Node* &root) {
	rRotate(root->r);
	lRotate(root);
}

void Insert(Node* &root, int val) {
	if(root == NULL) {
		root = new Node;
		root->data = val;
		//cout << "叶子结点无需调整即"<<val << endl;
	} else if(root->data > val) {
		//cout << val << "插入左子树" << root->data<< endl;
		Insert(root->l, val);

		if(getH(root->l)-getH(root->r) >= 2) {//左子树高度相比右子树高2,因为每次都只插入一个数，高度大于一就会调整，所以高度不会>2
			if(val <= root->l->data) {
			//	cout << "右旋" <<val<< endl;
				rRotate(root);
			//	cout << "结果" << root->data << endl;
			} else {
			//	cout << "左右旋" <<val<< endl;
				lrRotate(root);
			//	cout << "结果" << root->data << endl;

			}
		} //else cout << "左子树无需调整即"<<val << endl;

	} else {
		//cout << val << "插入右子树" << root->data<< endl;

		Insert(root->r, val);

		if(getH(root->r)-getH(root->l) >= 2) {//左子树高度相比右子树高2,因为每次都只插入一个数，高度大于一就会调整，所以高度不会>2
			if(val > root->r->data) {
		//		cout << "左旋" <<val<< endl;
				lRotate(root);
		//		cout << "结果" << root->data << endl;

			} else {
			//	cout << "右zuo旋" <<val<< endl;
				rlRotate(root);
			//	cout << "结果" << root->data << endl;

			}
		} //else cout << "右子树无需调整即"<<val << endl;
	}

}


void Order(Node *root, int idx) {
	if(root == NULL) return;
	if(idx > n) flag = false;

	level[idx] = root->data;
//	cout << idx << "对应"<< root->data << endl; 
	Order(root->l, idx*2);
	Order(root->r, idx*2+1);
}
int main() {
	cin >> n;

	Node *root = NULL;

	for(int i=0; i<n; i++) {
		int tmp;

		cin >> tmp;

		Insert(root, tmp);
	}

	Order(root, 1);
	int i=1;
	while(n) {
		if(level[i]) { 
			cout << level[i];
			n--;
			if(n) cout << " ";
			else cout << endl; 
		}
		i++;
	}
	printf("%s\n", flag?"YES" : "NO");
	return 0;
}
/*
5
88 70 61 63 65
*/