从前序与中序遍历序列构造二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

主要思想：前序遍历为根左右，所以第一个为根节点，中序遍历为左根右，可以在中序数组中找到根节点，记录长度，方便在前序数组中递归调用，然后左面为左子树，右面为右子树，递归构建二叉树。

/**
     * 前序和中序构建树
     * @param preorder 前序遍历数组
     * @param inorder 中序遍历数组
     * @return 构建树的根
     */
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder.length==0) return null;
        return helper(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
    }

    /**
     * 构建二叉树
     * @param preoder   前序遍历数组
     * @param preLeft   前序左边界
     * @param preRight  前序右边界
     * @param inorder   中序遍历数组
     * @param inLeft    中序左边界
     * @param inRight   中序右边界
     * @return
     */
    public TreeNode helper(int[] preoder,int preLeft,int preRight,int[] inorder,int inLeft,int inRight){
//        说明没有子树可以构建，返回空
        if (preLeft>preRight) return null;
//        构建根节点，前序遍历第一个节点就是根节点
        TreeNode root = new TreeNode(preoder[preLeft]);
//        在中序遍历中寻找根节点，左边为左子树，右边为右子树
        int len = 0;
        for (int i=inLeft;i<=inRight;i++){
            if (preoder[preLeft]==inorder[i])
                break;
            len++;
        }
//        中序遍历可以找到根节点，左边是左子树，右边是右子树，递归构建二叉树
        root.left = helper(preoder,preLeft+1,preLeft+len,inorder,inLeft,inLeft+len);
        root.right = helper(preoder,preLeft+len+1,preRight,inorder,inLeft+len+1,inRight);
        return root;
    }