题目描述:
给定节点数为 n 的二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示。
主要思想:通过先序遍历的结果先确定根节点在中序遍历中的位置,之后将中序遍历分为左右两段对应的是根节点的左右子树,利用递归一层层执行,最终当区间没有值时返回。
public class PreOrderAndInOrder {
public TreeNode buildTree(int[] preorder, int[] inorder) {
return buildTreeHelper(preorder,inorder,0,inorder.length - 1);
}
/**
* 借助前序遍历,在中序遍历结果集[left...right]上还原二叉树
*/
// 当前处理到前序遍历的哪个结点了
int index = 0;
private TreeNode buildTreeHelper(int[] preorder, int[] inorder, int left, int right) {
// base case
if (index == preorder.length) {
// 所有节点已经处理完毕
return null;
}
if (left > right) {
// 当前中序遍历的[left..right] 区间中没有元素
return null;
}
// 1.先从前序遍历的结果集中找到根节点
int val = preorder[index];
index ++;
TreeNode root = new TreeNode(val);
// 2.再从中序遍历中找到当前树根的索引pos
int pos = find(inorder,val);
// 3.此时根据中序遍历的结果位置
root.left = buildTreeHelper(preorder,inorder,left,pos - 1);
root.right = buildTreeHelper(preorder,inorder,pos + 1,right);
return root;
}
// 在当前中序遍历的结果集[left..right]找到待查找元素val的索引
private int find(int[] inorder, int val) {
for (int i = 0;i < inorder.length;i++) {
if(inorder[i] == val) {
return i;
}
}
return -1;
}
}