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

题目

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

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

示例

例如，给出

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

返回如下的二叉树：

   3
   / \
  9  20
    /  \
   15   7

解题思路

1.求结点的个数
2.根据前序遍历求根结点（第一个）
3.在中序遍历中找到根结点
4.以根结点为中心把结点分别存入左右子树
5.左右子树递归遍历建立二叉树

代码实现

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

        int len = preorder.length;
        if (len == 0) {
            return null;
        }
        TreeNode root = new TreeNode(preorder[0]);
        for (int i = 0; i < len; i++) {
            
            if (inorder[i] == preorder[0]) {

                root.left = buildTree(Arrays.copyOfRange(preorder, 1, i+1), Arrays.copyOfRange(inorder, 0, i));
                root.right = buildTree(Arrays.copyOfRange(preorder, i+1, len), Arrays.copyOfRange(inorder, i+1, len));

                break;
                
            }
        }
        return root;

    }
}

复杂度分析

• 时间复杂度：$\mathcal{O}(N)$。
• 空间复杂度：$\mathcal{O}(N)$，存储整棵树。