题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解析:
已知树的先序遍历和中序遍历,需要我们重建这个二叉树,其实大家都知道三种顺序遍历,我们只用只要其中两个就可以重建这个二叉树了.拿到这道题先分析一下,不用代码实现,我们自己如果要重建这个二叉树在纸上我们会怎么做.
1. 找根节点
2. 找出左右子树
3. 在左右子树中继续找子树的根节点,直到节点为空,则二叉树重建完成
首先怎么找根节点,这很简单,先序遍历的第一个节点就是根节点,如果是后续遍历就倒过来,最后一个是根节点.怎么找左右子树呢?在中序遍历中找到根节点的位置,然后根节点左边的序列就是左子树的中序遍历序列.右边同理.接下来就是一个不断的递归完成子问题的过程,在以子树的根节点为根继续向下.
/**
* 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) {
TreeNode root = reConstructBinaryTree(pre , 0 , pre.length-1 , in,0,in.length-1);
return root;
}
private TreeNode reConstructBinaryTree(int[] pre ,int preStart , int preEnd , int[] in , int inStart , int inEnd){
if(preStart > preEnd || inStart > inEnd){
return null;
}
TreeNode root = new TreeNode(pre[preStart]);
for(int i = inStart ; i <= inEnd ; i++){
if(in[i] == pre[preStart]){
root.left = reConstructBinaryTree(pre , preStart+1 , preStart + i - inStart , in,inStart,i-1);
root.right = reConstructBinaryTree(pre , i-inStart+preStart+1,preEnd,in,i+1,inEnd);
}
}
return root;
}
}