package src.com.zhang.tree.avltree;
//import Node;
public class AVLTreeDemo {
public static void main(String[] args) {
int[] arr={10,12,8,9,7,6};
//创建一个AVL对象
AVLTree avlTree=new AVLTree();
//添加节点
for (int i=0;i<arr.length;i++){
avlTree.add(new Node(arr[i]));
}
avlTree.indexOrder();
System.out.println("在没有平衡处理之前:");
System.out.println(avlTree.getRoot().height());
System.out.println(avlTree.getRoot().leftHeight());
System.out.println(avlTree.getRoot().rightHeight());
}
}
class AVLTree {
private Node root;
public Node getRoot() {
return root;
}
public void setRoot(Node root) {
this.root = root;
}
public Node search(int value) {
if (root == null) {
return null;
} else {
return root.search(value);
}
}
public Node searchParent(int value) {
if (root != null) {
return null;
} else {
return root.serchParent(value);
}
}
//删除节点
public void delNode(int value) {
if (root == null) {
System.out.println("root为空");
return;
} else {
//先找到需要删除的节点
Node targetNode = this.search(value);
//如果没有找到,则停止
if (targetNode == null) {
System.out.println("11111111");
return;
}
//如果targetNode没有父节点
if (root.left == null && root.right == null) {
root = null;
return;
}
System.out.println("22222");
Node parent = root.serchParent(value);
//如果要删除的是叶子节点
if (targetNode.left == null && targetNode.right == null) {
//判断targetNode是parent的左子节点还是又子节点
if (parent.left != null && parent.left.value == value) {
System.out.println("33333333");
parent.left = null;
} else if (parent.right != null && parent.right.value == value) {
System.out.println("4444444");
parent.right = null;
}
} else if (targetNode.left != null && targetNode.right != null) //删除有两颗子树的节点
{
int min = delRightTreeMin(targetNode.right);
targetNode.value = min;
} else//删除只有一颗子树的节点
{
//如果要删除的节点有左子节点
if (targetNode.left != null) {
if (parent != null) {
if (parent.left.value == value) {
parent.left = targetNode.left;
} else {
parent.right = targetNode.left;
}
} else {
root = targetNode.left;
}
} else //如果要删除的节点有右子节点
{
if (parent != null) {
if (parent.left.value == value) {
parent.left = targetNode.right;
} else {
parent.right = targetNode.right;
}
} else {
root = targetNode.right;
}
}
}
}
}
public void add(Node node){
if (root==null){
root=node;
}
else {
root.add(node);
}
}
public void indexOrder(){
if (root==null){
System.out.println("空树不能遍历");
}else {
root.indexOrder();
}
}
public int delRightTreeMin(Node node){
Node target=node;
while (target.left!=null){
target=target.left;
}
delNode(target.value);
return target.value;
}
}
class Node{
int value;
Node left;
Node right;
/**
* 左旋转
*/
private void leftRotate(){
//1.创建一个新的节点,放入当前节点的值
Node newNode = new Node(value);
//2.把新的节点的左子树设置为当前节点的左子树
newNode.left=left;
//3.把新的节点的右子树设置为之前节点的右子树的左子树
newNode.right=right.left;
//4.把当前节点的值换成右子节点的值
value=right.value;
//5.把当前节点的右子树设置成当前节点的右子树的右子树
right=right.right;
//6.把当前节点的左子节点设置成新的节点
left=newNode;
}
private 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;
}
/**
* 返回左子树的高度
* @return
*/
public int leftHeight(){
if (left==null){
return 0;
}
return left.height();
}
/**
* 返回右子树的高度
* @return
*/
public int rightHeight(){
if (right==null){
return 0;
}
return right.height();
}
//返回当前以当前节点为根节点的树的高度
public int height(){
return Math.max(left==null?0: left.height(), right==null?0:right.height())+1;
}
//查找要删除的节点
public Node search(int value){
if (value==this.value){
System.out.println("xxxxxxxxx");
return this;
}
else if(this.value>value){
if (this.left==null){
System.out.println("xxxxxxxxx1");
return null;
}
System.out.println("wwwwwww1");
return this.left.search(value);
}else {
if (this.right==null){
System.out.println("xxxxxxxxx2");
return null;
}
System.out.println("wwwwwwwwww2");
return this.right.search(value);
}
}
/**
*
* @param value 要删除的节点的值
* @return
*/
//查找要删除节点的父节点
public Node serchParent(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.serchParent(value);
}else if (value>=this.value&&this.right!=null){
return this.right.serchParent(value);
}else {
return null;
}
}
}
public Node(int value){
this.value=value;
}
//构建二叉排序树
public void add(Node node){
if (node==null){
return;
}
if (node.value<this.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){
if (right!=null && right.leftHeight()>right.rightHeight()){
right.rightRotate();
}
leftRotate();
return;//这里的return必须加,否则可能会执行下面的if()
}
//判断是否需要右旋转
if (leftHeight()-rightHeight()>1){
//如果它的左子树的右子树的长度大于左子树,先左旋转再右旋转
if (left!=null && left.rightHeight()>left.leftHeight()){
left.leftRotate();
}
rightRotate();
}
}
//中序遍历
public void indexOrder(){
if (this.left!=null){
this.left.indexOrder();
}
System.out.println(this.value);
if (this.right!=null){
this.right.indexOrder();
}
}
}
java实现avl树
最新推荐文章于 2023-03-11 17:03:20 发布