剑指offer算法-----根据前序遍历和中序遍历重新生成二叉树

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

积沙成塔

题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

算法分析:

在根据前序遍历和中序遍历生成二叉树的情景中,首先通过前序遍历得到根,再根据中序遍历得到左右子树的元素,再按照同样的思路继续生成左右子树。从上边可以看出这是一个递归的过程,因此实现思路是在确定左右子树的前提下递归创建子树。

实现代码:

class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode(int x) { val = x; }
  }

public class CreatTree {
    public TreeNode reConstructBinaryTree(int [] pre, int [] in) {
            if(pre.length==1)
                return new TreeNode(pre[0]);
            if(pre.length==0)
                return null;
            int index=0;
            TreeNode gen=new TreeNode(pre[0]);
            while(in[index]!=pre[0])
                index++;//确定根
            int[] inLeft=new int[index];//确定左子树中序
            int[] preLeft=new int[index];//确定左子树前序
            for(int i=0;i<index;i++){
                inLeft[i]=in[i];
                preLeft[i]=pre[i+1];
            }
            int[] inRight=new int[in.length-index-1];//确定右子树中序
            int[] preRight=new int[in.length-index-1];//确定右子树前序
            for(int i=index+1;i<in.length;i++){
                inRight[i-index-1]=in[i];
                preRight[i-index-1]=pre[i];
            }
            gen.left=reConstructBinaryTree(preLeft,inLeft);
            gen.right=reConstructBinaryTree(preRight,inRight);
            return gen;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_40400960/article/details/81866378