Modificar un BinarySearchTree que ser equilibrada (AVL): Java

votos
2

Necesito modificar un árbol binario de búsqueda que he creado para asegurar que esté equilibrado. Sólo tengo que modificar el añadir y eliminar métodos, de acuerdo con mis instrucciones. Esto es lo que tengo actualmente:

package proj;

public class BinarySearchTree<T extends Comparable<T>>{
    public static void main(String[] args) {
        BinarySearchTree<Integer> tree = new BinarySearchTree<Integer>();
        tree.add(5);
        tree.add(1);
        tree.add(2);
        tree.add(6);
    }

    private Node<T> root;
    private int size;
    String inorder = ;
    String preorder = ;

    public BinarySearchTree(){
        root = null;
        size = 0;
    }

    //adds a new item to the queue
    public void add(T obj) {
        Node<T> n = new Node<T>(obj);
        if( root == null ) {
            root = n;
        } else {
            add( root, n );
        }
        size++;
    }

    private void add(Node<T> subtree, Node<T> n) {
        if( subtree.getValue().compareTo(n.getValue()) > 0 ) {
            if( subtree.getLeftChild() == null ) {
                subtree.setLeftChild(n);
                n.setParent(subtree);
            } else {
                add( subtree.getLeftChild(), n );
            }
        } else {
            if( subtree.getRightChild() == null ) {
                subtree.setRightChild(n);
                n.setParent(subtree);
            } else {
                add( subtree.getRightChild(), n );
            }
        }
    }

    //returns the head of the queue
    public T peek(){
        Node<T> current = root;
        while(current.getLeftChild() != null){
            current = current.getLeftChild();
        }
        return current.getValue();
    }

    //removes the head of the queue and returns it
    public T remove(){
        if(root == null){
            return null;
        }

        Node<T> current = root;
        while(current.getLeftChild() != null){
            current = current.getLeftChild();
        }
        if( current.getParent() == null ) {
            root = current.getRightChild();
            if(root != null){
                root.setParent(null);
            }
        } else {
            current.getParent().setLeftChild(current.getRightChild());
            if(current.getRightChild() != null){
                current.getRightChild().setParent(current.getParent());
            }
        }
        size--;
        return current.getValue();
    }

    //returns the position of an element in the queue, or -1 if it is not found
    public int search(T searchItem){
        String tempOrdered = inorder(root);
        for(int i = 0; i<tempOrdered.length(); i++){
            if(String.valueOf(tempOrdered.charAt(i)).equals(searchItem.toString())){
                return i;
            }
        }
        return -1;
    }

    //returns number of nodes in the tree
    //returns the total number of elements in the queue
    public int getSize(){
        return size;
    }
    public String inorder() {
        inorder = ;
        if( root == null )
            return inorder;
        return inorder(root);
    }

    //returns an in-order, comma-separated string of every element in the queue
    private String inorder(Node<T> n){
        if(n.getLeftChild() != null){
            inorder(n.getLeftChild());
        }
        inorder += n.getValue();
        if(n.getRightChild() != null){
            inorder(n.getRightChild());
        }
        return inorder;
    }

    public String preorder() {
        preorder = ;
        if( root == null )
            return preorder;
        return preorder(root);
    }

    //returns a pre-ordered, comma-separated string of every element in the queue
    private String preorder(Node<T> n){
        preorder+= n.getValue();
        if(n.getLeftChild() != null){
            preorder(n.getLeftChild());
        }
        if(n.getRightChild() != null){
            preorder(n.getRightChild());
        }

        return preorder;
    }

    //returns the height of the tree; returns -1 if the tree is empty
    public int height(Node<T> n){
        if(n == null){
            return -1;
        }
        return Math.max(height(n.getLeftChild()), height(n.getRightChild()))+ 1;
    }

    //returns the root node
    public Node<T> getRoot(){
        return root;
    }
}

No estoy buscando a alguien para caminar yo con esta asignación - simplemente buscando algún consejo en cuanto a cómo debería ir haciendo esto para que no rompo la mitad del camino en el código que supongo que voy a necesitar. hacer algo al efecto de comprobar el factor de equilibrio del árbol se añade o elimina cada vez que algo, entonces reconstruir el árbol o 'rotar' cuando está desequilibrado.

Gracias por cualquier consejo dado con antelación. :) apreciar todas las puntas.

-Chris

Publicado el 02/06/2011 a las 17:45
fuente por usuario
En otros idiomas...                            


1 respuestas

votos
1

El árbol AVL artículo sobre Wikipedia da todo lo que necesita para poner en práctica este tipo de árbol de auto-equilibrado (me gusta especialmente la imagen que muestra las rotaciones necesarias para el reequilibrio). Básicamente es necesario implementar la rotación del árbol de izquierda y derecha y utilizarlo en sus addy removeprocedimientos de acuerdo con las reglas establecidas en el artículo.

Si usted es más aventurero, pruebe la implementación de un árbol rojo-negro. Una buena descripción con pseudo código se puede encontrar en Introducción a los algoritmos .

Respondida el 02/06/2011 a las 17:56
fuente por usuario

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more