LeetCode 刷题系列 105 前序与中序遍历序列构造二叉树

给定一棵树的前序遍历 preorder 与中序遍历  inorder。请构造二叉树并返回其根节点。

示例 1:

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]
 

提示:

1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder 和 inorder 均无重复元素
inorder 均出现在 preorder
preorder 保证为二叉树的前序遍历序列
inorder 保证为二叉树的中序遍历序列

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal
 

思路：

• 此题可以用递归来解决，设有一个函数为 buildTreeHelper( int[] preorder, int p_start, int p_end, int[] inorder, int i_start, int i_end ) , 该函数可以将前序遍历 preorder 与 中序遍历inorder 构建成树

1. 前序遍历为： 根节点，[左子树集合]，[右子树集合]
2. 中序遍历为：[左子树集合]，根节点，[右子树集合]
3. 我们可以先从 1 preorder 中找到 根节点 对应的 val 
4. 再 从 2 inorder中获取根节点左子树集合与右子树的长度，由此分别得到

           4.1    根节点左子树的前序遍历集合，根节点左子树的中序遍历集合

           4.2    根节点右子树的前序遍历集合，根节点右子树的中序遍历集合

      5.   4.1 与 4.2 得到的两个集合 放入递归函数 buildTreeHelper 调用，分别得到左子树集合生成的树 与 右子树集合生成的树

   注意：既然是递归函数，要有递归终止条件。

       如果 p_end 与 i_end 分别代表 preorder 与 inorder 的最后一个元素，那么，递归终止条件为：  p_start > p_end or i_start > i_end

       如果 p_end 与 i_end 分别代表 preorder 与 inorder 的最后一个元素 的后一个元素，那么，

        递归终止条件为：  p_start == p_end or i_start  == i_end         

Java代码如下：

/**
 * 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) {
            if (preorder.length==0) {
                return null;
            }

            return buildTreeHelper(preorder,0,preorder.length-1,inorder,0,inorder.length-1);

    }

    TreeNode buildTreeHelper(int[] preorder, int p_start, int p_end, int[] inorder, int i_start, int i_end) {
            if(p_start > p_end) {
                return null;
            }
            
            int rootVal = preorder[p_start];
            TreeNode root = new TreeNode(rootVal);

            int root_i_index=0;
            for (int i=i_start;i<=i_end;i++) {
                    if(inorder[i] == rootVal) {
                        root_i_index = i;
                        break;
                    }
            }

            root.left = buildTreeHelper(preorder,p_start+1,p_start + root_i_index-i_start,inorder,i_start,root_i_index-1);
            root.right = buildTreeHelper(preorder,p_start + root_i_index-i_start+1,p_end,inorder,root_i_index+1,i_end);

            return root;
    }

}