1123 Is It a Complete AVL Tree (30分)

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

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

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

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

图示理解

#include
#include
#include
#include

using namespace std;
vector 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 << "叶子结点无需调整即"<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 << "右旋" <data << endl;
			} else {
			//	cout << "左右旋" <data << endl;

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

		Insert(root->r, val);

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

			} else {
			//	cout << "右zuo旋" <data << endl;

			}
		} //else cout << "右子树无需调整即"< 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> 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
*/

 

你可能感兴趣的:(PAT,二叉树)