题目描述:
给定节点数为 n 的二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示。
示例:
输入:
[1,2,4,7,3,5,6,8],[4,7,2,1,5,3,8,6]
返回值:
{1,2,3,4,#,5,6,#,7,#,#,8}
说明:
返回根节点,系统会输出整颗二叉树对比结果,重建结果如题面图示
算法思路:
对于给定一棵二叉树的前序遍历和中序遍历,对于前序遍历是按照“根左右”、中序遍历是按照“左根右”的次序进行的,那么二叉树的根其实就是前序遍历的第一个元素,然后对于该二叉树的左右子树,在中序遍历中找,对应根节点的索引左侧所有元素是左子树中的节点,右侧所有元素是右子树的节点,然后依次递归进行前序和中序遍历构造左右子树,直到到达叶子节点。
算法实现:
import java.util.*;
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
//pre表示前序,vin表示中序
if (pre.length==0&&vin.length==0){
return null;
}
//前序遍历的第一个节点就是根节点
TreeNode root = new TreeNode(pre[0]);
//找到左右子树
for (int i=0;i<vin.length;i++){
if (vin[i] == root.val){
//该处左侧为左子树,右侧为右子树
root.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(vin,0,i));
root.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(vin,i+1,vin.length));
}
}
return root;
}
}