题目
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
注意:
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例 1:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
限制:0 <= 节点个数 <= 5000
思路
1、前序遍历顺序为:根节点 --> 左节点–> 右节点,则为【根节点,左子树, 右子树】;
2、中序遍历顺序为:左节点 --> 根节点 --> 右节点,则为【左子树,根节点, 右子树】;
3、因为前序遍历第一个元素为根节点,中序遍历的根节点左边为左子树,右边为右子树,所以我们根据二叉树的前序遍历和中序遍历可以划分出二叉树的根节点、左子树和右子树;
4、划分出的左子树、右子树的前序遍历和中序遍历又可以划分出对应二叉树的根节点、左子树和右子树,则可以使用递归重构二叉树。
注意点
1、使用映射(map)记录中序遍历节点和位置(下标)的映射关系,使因为树不含重复的数字,构建映射便于在递归过程中快速定位根节点的在中序遍历中的位置;(这样避免递归一次,遍历一次中序遍历寻找根节点的位置)
2、构建左子树时,因为前序遍历根节点的下一个节点就是左子树的根节点,则左子树的根节点在前序遍历中的位置为 root + 1;
3、构建右子树时,右子树的根节点在前序遍历中的位置为:根节点 + 左子树个个数,左子树个数 = 中序遍历根节点位置 - 开始遍历位置(left)+ 1,
则右子树的根节点在前序遍历中的位置为:root + (index - left + 1)。
实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }