第五题。。

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

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

分析：
前序遍历：根节点 左子树的前序遍历结果  右子树的前序遍历结果
中序遍历：左子树的中序遍历结果 根节点  右子树的中序遍历结果

方法一：

思路：①在前序遍历中取出根节点，得到其在中序遍历中的位置
②根据①可以知道左/右子树节点个数，于是可以得到左/右子树的前/中序遍历结果
③递归直到左/右子树为空

时间复杂度o(n)
空间复杂度o(n)

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //先做一个哈希表，便于定位根节点在中序遍历表中的位置(值，位置)
    static HashMap<Integer, Integer> map = new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = preorder.length;
        if (n == 0){
            return null;
        }
        //位置表
        for (int i = 0;i < n;i++){
            map.put(inorder[i],i);
        }
        TreeNode node = creatTree(preorder,inorder,0,n-1,0,n-1);
        return node;
    }
    //递归得到二叉树(前中序遍历表、子树的前中序边界)
    public static TreeNode creatTree(int[] preorder,int[] inorder,int preleft,int preright,int inleft,int inright){
        //获得根节点
        TreeNode node = new TreeNode(preorder[preleft]);
        //根节点位置
        Integer index = map.get(node.val);
        //左子树节点个数----定位左子树边界，右子树边界
        int count = index - inleft;
        //左节点(count = 0时，没有左子树，不需要考虑)
        if (count > 0){
            TreeNode left = creatTree(preorder, inorder, preleft + 1, preleft + count, inleft, index - 1);
            node.left = left;
        }
        //右节点(没有右节点时，也不用考虑)
        if (index < inright){
            TreeNode right = creatTree(preorder, inorder, preleft + count + 1, preright, index+1, inright);
            node.right = right;
        }
        return node;
    }
}

方法二：

迭代法（马马虎虎，看很久，似懂非懂，代码先放在这里）
 

package exercise05;

import java.util.Stack;

public class Solution2 {
    public static void main(String[] args) {
        int[] preorder = {3,9,10,15,11,17,12,14,18,20};
        int[] inorder = {15,10,9,11,17,3,14,12,20,18};
        buildTree(preorder,inorder);
    }
    public static TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = preorder.length;
        if (n == 0){
            return null;
        }
        //将根节点放入栈中（第一个要操作的节点）
        TreeNode node = new TreeNode(preorder[0]);
        Stack<TreeNode> stack = new Stack<>();
        stack.push(node);
        //从前序遍历中开始遍历：如果结点不是中序遍历中的要遍历的那一个节点，说明后面的结点是左节点
        int k = 0;//中序遍历的遍历索引
        for (int i = 1;i < preorder.length;i++){
            //peek代表要操作的节点（添加左节点或者右节点）
            TreeNode peek = stack.peek();
            TreeNode temp = new TreeNode(preorder[i]);
            if (peek.val != inorder[k]){//说明temp是左节点
                peek.left = temp;
                stack.push(temp);
            }else{//到达最左边，此时栈顶结点①和中序遍历结点①一致，此时同时出栈一个①，遍历下一个②，进行比较
                // 如果栈顶②和②仍然一致，说明①没有右节点，继续出栈直到栈顶与中序遍历不一致（假如是⑤）
                // 此时说明④有右节点，⑤是④右子树上的第一个中序遍历结点，那么temp就是④的右节点
                // 接着继续遍历：temp相当于刚开始的根节点
                while (!stack.isEmpty() && stack.peek().val == inorder[k]){//当不满足条件时，说明取出来的是有右节点的结点
                    peek = stack.pop();//遇到有右节点的结点就取出来进行下一步
                    k++;
                }
                peek.right = temp;
                stack.push(temp);
            }
        }
        return node;
    }
}