题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如:输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:二叉树的前序遍历时,第一个数字总是树的根节点的值。但在中序遍历中,根节点的值在序列的中间,左子树的结点的值位于根节点的值的左边,而右子树的结点的值位于根节点的右边。
所以 1.扫描中序遍历,找到根节点的索引,其前面的元素都是左子树的节点,右面的元素都是右子树的节点
2.递归
/**
* 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 [] in) {
if (pre == null || in == null) {
return null;
}
return constructCore(pre, 0, pre.length -1, in, 0, in.length -1);
}
public TreeNode constructCore(int [] preOrder,int startPreIndex,int endPreIndex,
int [] inOrder,int startInIndex,int endInIndex){
//创建根节点
int rootValue = preOrder[startPreIndex];
TreeNode root = new TreeNode(rootValue);
//只有一个元素
if (startPreIndex == endPreIndex) {
if (startInIndex == endInIndex &&
preOrder[startPreIndex] == inOrder[startInIndex]) {
return root;
}
}
//在中序遍历中找到根节点的索引
int rootIndex = startInIndex;
while (rootIndex <= endInIndex && inOrder[rootIndex] != rootValue) {
++rootIndex;
}
//左子树节点个数
int leftLength = rootIndex - startInIndex;
int leftPreOrderEndIndex = startPreIndex + leftLength;
if (leftLength > 0) {
//构建左子树
root.left = constructCore(preOrder, startPreIndex + 1, leftPreOrderEndIndex,
inOrder, startInIndex,rootIndex - 1);
}
if (leftLength < endPreIndex - startPreIndex) {
//右子树有元素,构建右子树
root.right = constructCore(preOrder, leftPreOrderEndIndex + 1, endPreIndex,
inOrder, rootIndex + 1, endInIndex);
}
return root;
}
}