算法练习- LeetCode 剑指 Offer 07. 重建二叉树

今日心情：无语言表

题目描述：

 LeetCode 剑指 Offer 07. 重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

---

解题代码：

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

    HashMap<Integer, Integer> map = new HashMap<>();
    int[] preorder;

    public TreeNode buildTree(int[] preorder, int[] inorder) {

        this.preorder = preorder;

        for(int i = 0; i < inorder.length;i++){
            map.put(inorder[i],i);
        }
        return process(0,0,inorder.length-1);
    }


    public TreeNode process(int root, int left, int right){

        if(left > right){return null;}

        TreeNode curr = new TreeNode(preorder[root]);

        int mIndex = map.get(preorder[root]);

        curr.left = process(root+1,left,mIndex-1);
        curr.right = process(root+mIndex-left+1,mIndex+1,right); 

        return curr;
    }
}

---

解题思路：

 首先重构二叉树肯定使用递归的方法进行构建，所以在于如何确定递归的条件。

（1）前序遍历数组的首位肯定是根节点，然后是左子树部分，然后是右子树部分；

（2）通过遍历中序遍历数组，在中序数组中找到根节点的位置；在中序数组中根节点的左半部分为左子树，根节点的右部分为右子树；

（3）通过找到中序数组中根节点的位置，确定左子树部分在前序遍历数组中的位置范围，然后遍历左边，以及右子树在前序遍历数组中的范围，然后遍历右子树。然后递归下去，直到 left 等于 right的时候返回空，停止递归。

---