資料結構--AVL樹的insert()的Java實現

標籤：資料結構   avl樹insert實現   

一個AVL樹是其每個節點的左子樹和右子樹的高度差最多差1的二叉尋找樹；AVL樹是一種最古老的平衡尋找樹

上代碼：

package com.itany.avlshu;public class AVLTree<T extends Comparable<?super T>>{    private static class AvlNode<T>    {        private int height;        private T element;        private AvlNode<T> left;        private AvlNode<T> right;        public AvlNode(T element)        {            this(element,null,null);        }        public AvlNode(T element,AvlNode<T> left,AvlNode right)        {            this.element=element;            this.left=left;            this.right=right;            height=0;        }    }    private int height(AvlNode<T> node)    {        return (node==null)?-1:node.height;    }    private int compare(T x, T element)    {        return x.compareTo(element);    }    private AvlNode<T> insert(T x,AvlNode<T> t)    {        if(t==null)            return new AvlNode<T>(x,null,null);//創造出來時 預設height=0        int compareResult=compare(x,t.element);        if(compareResult<0)        {            t.left=insert(x,t.left);            //做完了插入之後 立馬比較t的左右兒子的高度差是否等於2 若是 則進行相應旋轉 若不是那麼下一步  直接更新這個t的height值            //此時左邊兒子高度比較大            if(height(t.left)-height(t.right)==2)                            {                //下面分兩種旋轉情況 一種是單旋轉 另一種是雙旋轉                if(compare(x,t.left.element)<0)                    t=rotateWithLeftChild(t);                else                    t=doubleRotateWithLeftChild(t);            }        }        else if(compareResult>0)        {            t.right=insert(x,t.right);            //做完了插入之後 立馬比較t的左右兒子的高度差是否等於2 若是 則進行相應旋轉 若不是那麼下一步  直接更新這個t的height值            //此時右邊兒子高度比較大            if(height(t.right)-height(t.left)==2)            {                //下面分兩種旋轉情況 一種是單旋轉 另一種是雙旋轉                if(compare(x,t.right.element)<0)                    t=doubleRotateWithRightChild(t);                else                    t=rotateWithRightChild(t);            }        }        else            ;        t.height=Math.max(height(t.left), height(t.right))+1;//+1是加一個自己         return t;    }    private AvlNode<T> rotateWithRightChild(AvlNode<T> k1)    {        AvlNode<T> k2=k1.right;        k1.right=k2.left;        k2.left=k1;        k2.height=Math.max(height(k2.left),height(k2.right))+1;        k1.height=Math.max(height(k1.left),height(k1.right))+1;        return k2;    }    private AvlNode<T> doubleRotateWithRightChild(AvlNode<T> k3)    {        k3.right=rotateWithLeftChild(k3.right);        return rotateWithRightChild(k3);    }    //雙旋轉是由兩次單旋轉得來    private AvlNode<T> doubleRotateWithLeftChild(AvlNode<T> k3)    {        k3.left=rotateWithRightChild(k3.left);        return rotateWithLeftChild(k3);    }    private AvlNode<T> rotateWithLeftChild(AvlNode<T> k2)    {        AvlNode<T> k1=k2.left;        k2.left=k1.right;        k1.right=k2;        k2.height=Math.max(height(k2.left),height(k2.right))+1;        k1.height=Math.max(height(k1.left),height(k1.right))+1;        return k1;    }    }

資料結構--AVL樹的insert()的Java實現