二进制树使用数组

时间:2014-02-25 19:55:28

标签: java arrays binary-tree

我正在尝试从第一个数组中插入键并将它们排序到第二个数组中的二叉树中。我知道我的插入方法有问题,但我无法弄明白。任何帮助将不胜感激

import java.util.Arrays;

public class BSTreeArray {
static int startValues[]={50, 25, 75, 10, 15, 5, 53, 29, 79, 78, 111, 33};
int [] aTree;
public BSTreeArray(){
    for(int i=0; i<startValues.length;i++){
        insert(startValues[i]);
    }
    System.out.println("Pre-Order:");
    preOrder(aTree[0], "");
}


public void insert(int key){
    aTree=new int [100];

    if(aTree[0]==0){
        aTree[0]=key;
    }

    boolean add = false;
    int curIdx=0;

    while(!add){
        if(key<aTree[curIdx]){
            //go left
            if (aTree[curIdx*2+1]==0){
                aTree[curIdx*2+1]=key;
                add = true;
            }else{
                curIdx=curIdx*2+1;
            }
        }else{
            //go right
            if(aTree[curIdx*2+2]==0){
                aTree[curIdx*2+2]= key;
                add=true;
            }else{
                curIdx=curIdx*2+2;
            }
        }
    }
}

public void preOrder(int idx, String i){
    if(idx>aTree.length){
        return;
    }else{
        System.out.println(i+aTree[idx]);
        preOrder(((2*idx)+1), i+".");
        preOrder(((2*idx)+2), i+".");

    }
}

public static void main (String []args){
    BSTreeArray a=new BSTreeArray();
}
 }

当前输出:

Pre-Order:
0
.0
.0

期望的输出:

50

....25

........10

............5

............15

........29

............33

....75

........53

........79

............78

............111

1 个答案:

答案 0 :(得分:1)

我几乎立即看到的一件事是第一次插入将被添加到数组两次。

if(aTree[0]==0){
    aTree[0]=key;
}

boolean add = false;
int curIdx=0;

while(!add){ // add is false so the loop is entered
    if(key<aTree[curIdx]){ // aTree[curIdx] == key so false

    }else{
        if(aTree[curIdx*2+2]==0){ // true because arrays are initialized
                                  // with all elements 0

            aTree[curIdx*2+2]= key; // assign key a second time
            add=true;
        }
    }
}

这可能会污染所有其余的插入,因为如果它们位于第一个节点的右侧,它们将被跳过一个节点。

好像你应该这样做:

if(aTree[0]==0){
    aTree[0]=key;

    return; // <-
}

(或者我把方法的其余部分放在else块中。)

我编译了它并自己运行并注意到其他一些简单的错误。

您要在每次调用时创建一个新数组:

public void insert(int key){
    aTree=new int [100];

因此,您可以将new int[100]移至aTree的声明:

public class BSTreeArray {
    int [] aTree = new int[100]; // <-

您使用preOrder而不是aTree[0]致电0

preOrder(aTree[0], "");

应该是:

preOrder(0, "");

preOrder内,您有idx>aTree.length

if(idx>aTree.length){

这会产生一个越界异常,因为数组索引是0...length - 1所以它应该是>=

if(idx>=aTree.length){

最后,由于您使用0表示空节点,因此您也可以在preOrder中检查该节点:

if(aTree[idx] != 0)
    System.out.println(i+aTree[idx]);

之后输出为:

Pre-Order:
50
.25
..10
...5
...15
..29
...33
.75
..53
..79
...78
...111

作为建议,您可能希望将数组设为Integer[]而不是int[]。这意味着你的int值将被装箱,但它会让你使用null来表示一个空节点而不是0.另一种方法是使用long[]并使空值为某个值int无法存储1L << 63LLong.MIN_VALUE)。