AVL树
平衡二叉树这个是建立在排序二叉树的基础上的,加了一个条件就是每个结点的左右子树结点的高度不能超过2(即要小于等于1)
其余的代码实现情况和排序树一样
但add方法里面在添加完结点后需要平衡
add方法首先分四种情况,LL,RR,LR,RL
第一个L/R是指:加在根节点的左边/右边
第二个L/R是指:导致不平衡的叶子节点的父节点是左节点/右节点(例如:3是左,6是右)
LL: RR:
LR: RL:
LR/RL需要通过先对左子树/右子树进行左/右旋转转成LL/RR的情况,再对LL/RR进行右旋转/左旋转
代码实现:
左旋转:
总共分为6步:1)先创建一个和根节点相同的新节点
2)再将新节点的左节点设为左子树
3)将新节点的右节点设为右子树的左节点
4)将根节点的值设为右子树的值
5)将根节点的右节点设为右子树的右子树
6)将根节点的左子树设为新的节点
右旋转:
情况和左旋转类似:
也分为6步:
注意:在旋转前要先判断是LL还是LR(RR还是RL),
通过其根节点左右子树的高度来判断,如果是L,其根节点的左子树的左子树的高度大于根节点的左子树的右子树的高度,就是LL,小于就是LR
代码如下:
public class AVLTreeDemo {
public static void main(String[] args) {
AVLTree avlTree = new AVLTree();
int [] arr={10,11,7,6,8,9};
for (int i = 0; i <arr.length ; i++) {
avlTree.add(new Node(arr[i]));
}
System.out.println("平衡处理后==");
System.out.println("树的高度为:"+avlTree.hight());
System.out.println("树的左子树的高度为:"+avlTree.lefthight());
System.out.println("树的右子树的高度为:"+avlTree.righthight());
}
}
class AVLTree{
private Node root;
public Node getRoot() {
return root;
}
//添加结点
public void add(Node node){
if (root!=null){
root.add(node);
}else {
root=node;
}
}
//中序遍历
public void infixOrder(){
if (root!=null){
root.infixOrder();
}else {
System.out.println("树为空,无法遍历!!!");
}
}
//查找结点
public Node search(int value){
if (root==null){
return null;
}
return root.search(value);
}
//查找需要结点的父节点
public Node searchParent(int value){
if (root==null){
return null;
}
return root.searchParent(value);
}
//找到最小值的子树的结点的值,并删除
public int searchSubtreeMin(Node node){
Node target=node;
while (target.left!=null){
target=target.left;
}
int res=target.value;
del(target.value);
return res;
}
//删除结点
public boolean del(int value){
boolean res=true;
Node targetNode = search(value);
if (targetNode==null){
return false;
}
Node targetParentNode = searchParent(value);
if (targetParentNode==null){//当父节点为空就判断删除的是不是根节点,不是就直接返回false
if (root.value!=value){
return false;
}
}
//如果是删除的是叶子结点
if (targetNode.left==null&&targetNode.right==null){
if (targetParentNode!=null) {
//判断是父节点的左节点还是右节点
if (targetParentNode.left != null && targetParentNode.left.value == value) {
targetParentNode.left = null;
} else if (targetParentNode.right != null && targetParentNode.right.value == value) {
targetParentNode.right = null;
}
}else {//如果父节点为空
root=null;
}
}else if (targetNode.left!=null&&targetNode.right != null){
int min = searchSubtreeMin(targetNode.right);
targetNode.value=min;
}else {//删除只有一个结点的
if (targetNode.left!=null){//如果目标结点只有左子结点
if (targetParentNode!=null) {
if (targetParentNode.left.value== value) {//如果目标结点为父节点的左结点
targetParentNode.left = targetNode.left;
} else if (targetParentNode.right.value==value){//如果目标结点为父节点的右结点
targetParentNode.right = targetNode.left;
}
}else {//如果删除的是根结点
root=root.left;
}
}else {//如果目标结点只有右子结点
if (targetParentNode!=null) {
if (targetParentNode.right.value==value) {// 如果目标结点为右结点
targetParentNode.right=targetNode.right;
}else if (targetParentNode.left.value==value){//如果目标结点为左结点
targetParentNode.left=targetNode.right;
}
}else {
root=root.right;
}
}
}
return res;
}
//查看树的高度
public int hight(){
if (root==null){
return 0;
}
return root.height();
}
//查看根的左子树的高度
public int lefthight(){
if (root==null){
return 0;
}
return root.leftheight();
}
//查看根的右子树的高度
public int righthight(){
if (root==null){
return 0;
}
return root.rightheight();
}
}
class Node{
int value;
Node left;
Node right;
public Node(int value) {
this.value = value;
}
@Override
public String toString() {
return "value: "+value;
}
//二叉排序树添加方法,递归的方法实现
public void add(Node node){
//先比较值的大小
if (this.value>node.value){
if (this.left==null){//如果左节点为空,就直接加上就行
this.left=node;
}else{//如果左结点不为空,递归
this.left.add(node);
}
}else {
if (this.right==null){
this.right=node;
}else {
this.right.add(node);
}
}
//在添加完之后,可以判断是否添加的不平衡,是否需要旋转
if (rightheight()-leftheight()>1){
//当添加的结点的根结点的右边
//则需要判断是RL还是RR
//通过根结点右子树的左右子树的高度来判断
if (right!=null&&right.leftheight()>right.rightheight()){//RL
//先对右结点进行右旋转,将其变成RR的情况
right.rightRotate();
//再对根节点进行左旋转
}
leftRotate();
// return;//!!!这个一定要写
}
if (leftheight()-rightheight()>1){
//当添加的结点在根节点的左边
//则需要判断是LR还是LL
//通过左子树的左右两边的高度来判断
if (left != null && left.rightheight() > left.leftheight()){//LR
//先对左子树进行左旋转将其变成LL的情况,再右旋转
left.leftRotate();
}
//LL情况就只要对其进行右旋转就行
rightRotate();
}
}
//中序遍历
public void infixOrder(){
if (this.left!=null){
this.left.infixOrder();
}
System.out.println(this);
if (this.right!=null){
this.right.infixOrder();
}
}
//查找结点
public Node search(int value){
if (this.value==value){
return this;
}else if (this.left!=null&&this.value>value){
return this.left.search(value);
}else if (this.right!=null&&this.value<value){
return this.right.search(value);
}else {
return null;
}
}
//查找的父节点
public Node searchParent(int value) {
//如果当前结点就是要删除的结点的父结点,就返回
if ((this.left != null && this.left.value == value) ||
(this.right != null && this.right.value == value)) {
return this;
} else {
//如果查找的值小于当前结点的值, 并且当前结点的左子结点不为空
if (value < this.value && this.left != null) {
return this.left.searchParent(value); //向左子树递归查找
} else if (value >= this.value && this.right != null) {
return this.right.searchParent(value); //向右子树递归查找
} else {
return null; // 没有找到父结点
}
}
}
//查找节点的高度
public int height(){
return Math.max(left == null ? 0 : left.height(),
right == null ? 0 : right.height())+1;
}
//查找左子树结点的高度
public int leftheight(){
if (left==null){
return 0;
}
return left.height();
}
//查找右子树结点的高度
public int rightheight(){
if (right==null){
return 0;
}
return right.height();
}
//左旋转
public void leftRotate(){
//1)首先创建新的根结点
Node node = new Node(value);
//2)将新结点的左结点设为左子树
node.left=left;
//3)将新结点的右结点设为右子树的左子树
node.right=right.left;
//4)将结点的值改为右结点的值
value=right.value;
//5) 将右子树设为右子树的右子树
right=right.right;
//6) 将新结点设为左子树
left=node;
}
//右旋转
public void rightRotate(){
//1) 首先创建新结点
Node newnode = new Node(value);
//2) 将新结点的右子树设为右子树
newnode.right=right;
//3) 将新结点的左子树设为左子树的右子树
newnode.left=left.right;
//4) 将结点的值设为左子树的值
value=left.value;
//5) 将左结点设为左子树的左子树
left=left.left;
//6) 将右子树设为新结点
right=newnode;
}
}