java 重建二叉树(前序和中序)

案例:给定两个数组,一个为二叉树的前序遍历,另一个为二叉树的中序遍历,还原二叉树

前序的作用就是确定根节点,中序可以根据前序的根节点确定该根节点的左右节点;

使用递归的方法实现

public class RebuildBinaryTree{
    // 二叉树数据结构
    public static class Tree{
    int val;
    Tree left;
    Tree right;
}
    public static void main(String[] agrs)
    {
        int[] prvOrder = {1,2,4,7,3,5,6,8};// 前序数组
        int[] inOrder = {4,7,2,1,5,3,8,6}; // 中序数组
        checkTree(prvOrder,inOrder);  // 检查输入合法性
    }
    public static void checkTree(int[] prvOrder,int[] inOrder)
    {    
        if(prvOrder==null||inOrder==null||prvOrder.length!=inOrder.length)
        {
            System.out.println("input error");
        }else
        {
            Tree head = getRebuild(prvOrder,0,prvOrder.length-1,inOrder,0,inOrder.length-1);
        }
    }
     /**
      * 根据前序遍历和中序遍历构建二叉树
      * @param preOrder    前序遍历序列
      * @param ps        前序遍历开始位置
      * @param pe        前序遍历结束位置
      * @param inOrder    中序遍历序列
      * @param is        中序遍历开始位置
      * @param ie        中序遍历结束位置
      * @return            构建的树的根节点
      */
    public static Tree getRebuild(int[] prvOrder,int ps,int pe,int[] inOrder,int is,int ie)
    {
        // 开始位置大于结束位置说明已经出来到叶节点
        if(ps>pe)
            return null;
        Tree root = new Tree();
        // 前序遍历的第一个为当前的根节点
        root.val = prvOrder[ps];
        // index 为根节点在中序遍历序列中的位置,为了下面计算出左右根节点左右子树的个数和起始位置
        int index = is;
        while(is<=ie&&inOrder[index]!=prvOrder[ps]) // 遍历查找根节点在中序的位置
        {
            index++;
        }
        // 递归遍历左子树
        // 当前节点的左子树的个数为 index-is
        // 左子树对应的前序遍历的位置proOrder[ps+1,ps+index-is]
        // 左子树对应的中序遍历的位置inOrder[is,index-1]
        root.left = getRebuild(prvOrder,ps+1,ps+index-is,inOrder,is,index-1);
        // 递归遍历右子树
        // 当前节点对应右子树的个数为 ie-index
        // 右子树对应的前序遍历的位置 prv[ps+index-is+1,pe]
        // 右子树对应的中序遍历的位置 inOrder[index+1,ie]
        root.right = getRebuild(prvOrder,ps+index-is+1,pe,inOrder,index+1,ie);
        // 递归结束返回根节点
        return root;
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

memory_cood

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值