比较器不适用于二叉树

时间:2015-04-04 11:27:11

标签: java recursion binary-tree comparator

我有这个东西

Comparator.java

public interface Comparator<T> {
    public int compareTo(int num);
}

valueComparator.java

public class valueComparator implements Comparator<Tree.Node> {
    @Override
    public int compareTo(Tree.Node obj, int number) {
        if (obj.getDataNumber() == number) {
            return 0;
        }
        else if (obj.getDataNumber() < number) {
            return -1;
        }
        else return 1;
    }
}

Tree.java

public class Tree {
    public Node root;

    Tree() {
    }

    public static class Node {

        Node(int number, String str, boolean flag) {
            dataNumber = number;
            dataText = str;
            dataBool = flag;
        }

        public int getDataNumber() {
            return this.dataNumber;
        }

        public String getDataText() {
            return this.dataText;
        }

        public boolean getDataBool() {
            return this.dataBool;
        }

        public void setDataText(String text) {
            this.dataText = text;
        }

        public void isDataBool(boolean flag) {
            this.dataBool = flag;
        }

        Node left;
        Node right;
        private int dataNumber;
        private String dataText;
        private boolean dataBool;

    }

    public void binaryTree() {
        root = null;
    }

    public boolean search(int number) {
        return search(root, number);
    }

    valueComparator comp = new valueComparator();

    private boolean search(Node node, int number) {
        if (node == null) {
            return false;
        }
        if (comp.compareTo(node, number) == 0) {
            return true;
        }
        if (comp.compareTo(node, number) == -1) {
            return search(node.left, number);
        }
        else {
            return search(node.right, number);
        }
    }  

    public void insertLeaf(int number, String str, boolean flag) {
    root = insertLeaf(root, number, str, flag);
    }
    private Node insertLeaf(Node node, int number, String str, boolean flag) {
        if (node == null) {
            node = new Node(number, str, flag);
        } else {
            if (number < node.dataNumber) {
                node.left = insertLeaf(node.left, number, str, flag);
            }
            else if (number > node.dataNumber) {
                node.right = insertLeaf(node.right, number, str, flag);
            }
            else {
                System.out.println("The element is already in the tree.");
            }
        }
        return node;
    }
}

Test.java

public class Test {
    public static void main(String args[]) {

        Tree binTree = new Tree();
        binTree.binaryTree();
        binTree.insertLeaf(5, "text2", true);
        binTree.insertLeaf(4, "text4", false);
        binTree.insertLeaf(1, "text1", true);
        binTree.insertLeaf(3, "text3", true);
        binTree.insertLeaf(2, "text5", false);
        System.out.println("Element 3 found: " + binTree.search(3));
    // Element 3 found: false
    }
}

我应该用比较器进行搜索,但我不理解逻辑。 compareTo方法适用于自身,但它停留在搜索的递归调用上。在第一次传递之后,如果compareTo的返回值不是= 0,那么它将以null进入并且中断递归并返回false。这意味着如果我将树的第一个元素设置为&#39; 3&#39;,则搜索(3)将返回true,但如果它不同于3 - false并且甚至看不到因为它在树上。

1 个答案:

答案 0 :(得分:1)

当您插入一个数字时,您可以将其直接与节点进行比较。值,如果数字小于当前节点中存储的值,则按照left指针。

但是当你搜索一个数字时,你会使用一个比较器,它将节点的值与给定的数字进行比较(注意相反的顺序!)如果数字少于那么当前节点中的值,您遵循right链接。

根据需要使用直接比较或比较器 - 但在任何地方都使用相同的方法。