重建二叉树

题目

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

解题

前序遍历：根、左、右
中序遍历：左、根、右
利用前序遍历，找到根结点，根据根结点的值在中序遍历中找到左右子树

/** * Definition for binary tree * public class TreeNode { *     int val; *     TreeNode left; *     TreeNode right; *     TreeNode(int x) { val = x; } * } */public class Solution {    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {        if(pre.length != in.length){            return null;        }        int prei = 0;        int prej = pre.length - 1;        int ini = 0;        int inj = in.length - 1;        return getTree(pre,prei,prej,in,ini,inj);    }    public TreeNode getTree(int[] pre,int prei,int prej,int[] in,int ini,int inj){        int val = pre[prei];// 根结点的值        TreeNode root = new TreeNode(val);        root.left = null;        root.right = null;        if(prei == prej)            return root;        int id = getRootId(in,ini,inj,val);// 根结点在中序遍历数组中的id        if(id == -1){            return null;        }        int leftLen = id-1 - ini + 1;// 左子树元素个数        int rightLen = inj - (id + 1) + 1;// 右子树元素个数        if(leftLen>0)            root.left = getTree(pre,prei + 1,prei + leftLen,in,ini,id-1);        if(rightLen>0)            root.right = getTree(pre,prei + leftLen + 1,prej,in,id + 1,inj);        return root;    }    public int getRootId(int[] in,int i,int j,int val){        for(int k = i;k<=j;k++){            if(in[k] == val){                return k;            }        }        return -1;    }}