1. 考察知识点
- 二叉树、递归
2. 题目
- 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
3. 思路
前序遍历:头左右,头结点为起始节点
中序遍历:左头右,头结点左边是左子树,右边是右子树
- 本题其实就是创建好一个二叉树,第一个元素为前序遍历的头结点,考虑左右子树的范围即可
- 只要在中序遍历中找到头结点的位置,就可以确定左右子树
4. 代码实现
/**
* 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) {
return reConstructBinaryTree(pre, 0, pre.length-1, in, 0, pre.length-1);
}
private TreeNode reConstructBinaryTree(int[] pre, int startPre, int endPre, int[] in, int startIn, int endIn) {
//异常情况
if (startPre > endPre || startIn > endIn)
return null;
//创建一个二叉树存储重建后的二叉树,该二叉树的头结点为startPre
TreeNode root = new TreeNode(pre[startPre]);
//因为前序遍历可以确定第一个数位头结点,遍历中序,找到中序遍历中头结点的位置,可以将遍历分成左二叉树和右二叉树两个部分
for (int i = startIn; i < in.length; i++) {
if (in[i] == pre[startPre]){
root.left = reConstructBinaryTree(pre, startPre+1, i-startIn+startPre, in, startIn, i-1);
root.right = reConstructBinaryTree(pre, i-startIn+startPre+1, endPre, in, i+1, endIn);
break;
}
}
return root;
}
}