Modifier un BinarySearchTree être équilibré (AVL): Java

voix
2

Je dois modifier un arbre de recherche binaire que j'ai créé pour vous assurer qu'il est équilibré. Je ne dois modifier l'ajouter et supprimer des méthodes, selon mes instructions. Voici ce que j'ai actuellement:

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;
    }
}

Je ne cherche pas quelqu'un pour me promener à travers cette mission - tout simplement à la recherche de quelques conseils quant à la façon dont je le faire de sorte que je ne romps pas le code à mi-chemin dans je devine que je vais avoir besoin. de faire quelque chose à l'effet de vérifier le facteur d'équilibre de l'arbre chaque fois que quelque chose est ajouté ou supprimé, puis reconstruire l'arbre ou « Pivoter » quand il est déséquilibré.

Merci pour tous les conseils donnés à l'avance. :) Appréciez tous les conseils.

-Chris

Créé 02/06/2011 à 17:45
source utilisateur
Dans d'autres langues...                            


1 réponses

voix
1

L' arbre AVL article sur Wikipédia donne tout ce que vous devez mettre en œuvre ce type d'arbre auto-équilibré (J'aime particulièrement l'image montrant les rotations nécessaires à un rééquilibrage). Fondamentalement , vous devez mettre en œuvre la rotation de l' arbre à gauche et à droite et l' utiliser dans vos addet removeméthodes selon les règles énoncées dans l'article.

Si vous êtes plus aventureux, essayez la mise en œuvre d' un arbre rouge-noir. Une bonne description avec le code pseudo se trouve dans Introduction aux algorithmes .

Créé 02/06/2011 à 17:56
source utilisateur

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