重建二叉树

来源:互联网 发布:男士护肤 知乎 旁氏 编辑:程序博客网 时间:2024/05/21 06:26

题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树

 /*
    解题思路:使用递归的思想,根据前序序列的节点,找出该节点在中序中的位置,并记录下
    该位置,然后生成节点,继续递归,直到前序序列的起始节点大于末尾节点时退出递归
    */
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        return buildTree(pre,in,0,pre.length-1,0,pre.length);
    }
    /*创建一个棵树,参数依次表示:前序序列,中序序列,前序序列的起始节点,
    前序列的长度减去1,中序序列的起始点,中序列的长度*/
    public TreeNode buildTree(int[] pre,int [] in,int pre_s,int pre_e,int in_s,int in_e){
        if(pre_s>pre_e){
            return null;
        }//出口条件
        int pivot=pre[in_s];
        int i=pre_s;
        for(;i<pre_e;i++){
            if(in[i]==pivot){//找出前序中的节点在中序中的位置,并且保存记录该位置的i
                break;
            }
        }//for
        TreeNode node=new TreeNode(pivot);//生成一个节点
        node.left=buildTree(pre,in,pre_s,i-1,in_s+1,i-pre_s+in_s);//以根节点划分后,左子树
        node.right=buildTree(pre,in,i+1,pre_e,i-pre_s+in_s+1,in_e);//以根节点划分后,右子树
        return node;
            
    }