Java创建二叉树

java 算法创建二叉树源码，转载

//定义一个节点
class BitNode {
    int data;
    BitNode lchild;
    BitNode rchild;

    public void setNode(int data, BitNode lc, BitNode rc) {
        this.data = data;
        lchild = lc;
        rchild = rc;
    }
}

static int counter = 0;//定义一个静态计数变量

/**
     * 构造<a href="https://beatree.cn/tag/%e4%ba%8c%e5%8f%89%e6%a0%91" title="查看更多关于二叉树的文章" target="_blank">二叉树</a>
     * 
     * @param root 根节点
     * @param a 数据源
     * @param i 计数器
     * @return 根节点
     */
    public static BitNode createBiTree(BitNode root, int[] a, int i) {
        if (i < a.length) {
            if (a[i] == 0) {  //以零作为分割，故开头不可是 0，否则整个树都为空
                root = null;
            } else {
                BitNode lchild = new BitNode();
                BitNode rchild = new BitNode();
                root.data = a[i];
                root.lchild = createBiTree(lchild, a, ++counter);
                root.rchild = createBiTree(rchild, a, ++counter);
            }
        }
        return root;
    }

// 访问节点
    public static void visitTNode(BitNode node) {
        System.out.print(node.data + " ");
    }

// 层次遍历
    public static void levelTraverse(BitNode root) {
        Queue queue = new LinkedList();
        queue.offer(root);// 从根节点入队列
        while (!queue.isEmpty()) {// 在队列为空前反复迭代
            BitNode bitNode = queue.poll();// 取出队列首节点
            visitTNode(bitNode);
            if (bitNode.lchild != null)
                queue.offer(bitNode.lchild);// 左孩子入列
            if (bitNode.rchild != null)
                queue.offer(bitNode.rchild);// 右孩子入列
        }
    }

public static void main(String[] args) {
        BitNode root = new BitNode();
        int[] a = { 1, 2, 3, 0, 0, 4, 0, 0, 5, 0, 0 };
        root = createBiTree(root, a, counter);
levelTraverse(root);
}

喜欢 (0) 赏

[欢迎投币]

分享 (0)