根据中序和前序序列生成二叉树java递归实现

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/wlittlefive/article/details/80463657

首先说下思路:

先给两个序列:

前序序列:1,2,4,8,5,3,6,7

中序序列:8,4,2,5,1,6,3,7

图脑补下吧偷笑 下面来分析下:

一、前序序列中的第一个肯定是根节点,知道了1是根节点后,我们去中序序列中找到1。

二、中序序列中1前面的 8,4,2,5 这4个肯定是左子树,6,3,7是右子树。

接下来关键的一步要想明白,我们要找到前序序列中的左子树的部分(这部分就是左子树的前序序列找到了我们就能递归的下去了)。由于我们在中序序列中知道了左子树有4个节点。那么我们根据前序序列的特性就可以得知。在根节点1 后面的4个节点就是我们的左子树了

综上可以得到:

左子树的前序序列是:2,4,8,5

左子树的中序序列是:8,4,2,5

重复上面的步骤:2是左子树的根节点,左子树的左子树的前序:4,8,左子树的中序序列是:8,4

右子树的前序序列是:3,6,7

右子树的中序序列是:6,3,7

重复上面的步骤:3是右子树的根节点,右子树的左子树的前序为:6,右子树的左子树的中序:6

上面的我们通过代码用递归很简单就能实现了:

上代码:

public class Tree {

    private int data;
    private Tree left;
    private Tree right;

    public Tree(final int data) {
        this(data, null, null);
    }

    public Tree(final int data, final Tree left, final Tree right) {
        this.data = data;
        this.left = left;
        this.right = right;
    }

    public static void main(String args[]) {
        int[] preOrders = new int[] {1, 2, 4, 8, 5, 3, 6, 7};
        int[] inOrders = new int[] {8, 4, 2, 5, 1, 6, 3, 7};
        buildTree(preOrders, inOrders);
    }

    private static Tree buildTree(int[] preOrders, int[] inOrders) {
        if (preOrders.length == 0 || inOrders.length == 0) {
            return null;
        }
        Tree tree = new Tree(preOrders[0]);
        int index = search(0, inOrders.length, inOrders, tree.getData());
        tree.setLeft(buildTree(Arrays.copyOfRange(preOrders, 1, index + 1), Arrays.copyOfRange(inOrders, 0, index)));
        tree.setRight(buildTree(Arrays.copyOfRange(preOrders, index + 1, preOrders.length),
            Arrays.copyOfRange(inOrders, index + 1, inOrders.length)));
        return tree;
    }

    private static int search(int start, int end, int[] inOrders, int data) {
        for (int i = start; i < end; i++) {
            if (data == inOrders[i]) {
                return i;
            }
        }
        return -1;
    }

}

省略了setter/getter。想了一会。转载希望注明出处。

猜你喜欢

转载自blog.csdn.net/wlittlefive/article/details/80463657