leetcode 剑指 Offer 07. 重建二叉树の解题思路

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出

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

    3
   / \
  9  20
    /  \
   15   7

解题思路

对二叉树缺少了解，导致拿到题后一脸懵逼，于是现在本子上画了一个三层的二叉树，然后用Root、Root.left、Root.left.left这种方式将前序遍历和中序遍历写一遍，然后开始找规律。

前序：R {R.l} {R.r}
中序：{R.l} R {R.r}

这时候发现，前序遍历，根节点必然是数组的第一个，而中序遍历时，根节点正好在左右子节点中间。

所以，递归的方案就出来了，从前序遍历中找出根节点，左边子节点的两种遍历数组和右边子节点的遍历数组都能知道了。

这样一直递归，知道遍历结果长度为1，就证明没有更多的子节点了。

代码

 * 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) {

        if (preorder.length == 0 || preorder.length != inorder.length) {
            return null;
        }

        TreeNode root = new TreeNode(preorder[0]);
        if (preorder.length != 1) {
            int leftNum = 0;
            for (int i = leftNum; i < inorder.length; i++) {
                if (preorder[0] == inorder[i]) {
                    leftNum = i;
                    break;
                }
            }
            root.left = buildTree(Arrays.copyOfRange(preorder, 1, 1 + leftNum),
                    Arrays.copyOfRange(inorder, 0, leftNum));
            root.right = buildTree(Arrays.copyOfRange(preorder, 1 + leftNum, preorder.length),
                    Arrays.copyOfRange(inorder, 1 + leftNum, inorder.length));
        }
        return root;
    }
}