劍指offer算法-----根據前序遍歷和中序遍歷重新生成二叉樹

積沙成塔

題目描述：輸入某二叉樹的前序遍歷和中序遍歷的結果，請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。例如輸入前序遍歷序列{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;
    }
}