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

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

一、题目详情

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

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

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

	3
   / \
  9  20
    /  \
   15   7

二、递归法

先序遍历的顺序是根节点，左子树，右子树。中序遍历的顺序是左子树，根节点，右子树。

所以只需要根据先序遍历得到根节点，然后在中序遍历中找到根节点的位置，它的左边就是左子树的节点，右边就是右子树的节点。

生成左子树和右子树就可以递归的进行了。

而且也不需要真的把preorder 和 inorder 切分了，只需要用分别用两个指针指向开头和结束位置即可。注意下边的两个指针指向的数组范围是包括左边界，不包括右边界。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return buildBinaryTree(preorder,0,preorder.length,inorder,0,inorder.length);
    }

    public TreeNode buildBinaryTree(int[] preorder, int p_start, int p_end, int[] inorder,int i_start, int i_end){
        // preorder数组为空，直接返回 null
        if(p_start == p_end){
            return null;
        }

        //定义二叉树的头结点
        int root_val = preorder[p_start];
        TreeNode root = new TreeNode(root_val);
        //找到inorder中的头结点所在位置，则该位置左边的为左子树，右边的为右子树
        int root_temp = 0;
        for(int i = 0;i < i_end;i++){
            if(inorder[i] == root_val){
                root_temp = i;
                break;
            }
        }

        //对数组的左右子树进行分割
        int left_num = root_temp - i_start;

        //递归对左右子树进行构造
        root.left = buildBinaryTree(preorder, p_start + 1, p_start + left_num + 1, inorder, i_start, root_temp);
        root.right = buildBinaryTree(preorder, p_start + left_num + 1, p_end, inorder, root_temp + 1, i_end);

        return root;
    }
}