输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并输出它的后序遍历序列。
分类:二叉树
解法1:明白由先序遍历和中序遍历生成二叉树的过程就好了。首先找到先序遍历的第一个节点,就是根节点,然后在中序遍历中找到根节点。
此时,根节点左右两边,分别是左右子树的中序遍历。
再对于先序遍历,从第二个节点开始,到中序遍历中左子树的长度,就是左子树的先序遍历。
同理,右子树也是一样,这是一个递归过程。
/**
* 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 build(pre,in,0,pre.length-1,0,in.length-1);
}
public TreeNode build(int[] pre,int[] in,int pstart,int pend,int istart,int iend){
if(pstart>pend) return null;
int cur = pre[pstart];//根节点
int find = istart;
while(find<=iend){//在中序遍历中查找根节点
if(cur==in[find]) break;
else find++;
}
int len = find-istart;
TreeNode res = new TreeNode(cur);//创建根节点
res.left = build(pre,in,pstart+1,pstart+len,istart,find-1);//创建左子树
res.right = build(pre,in,pstart+len+1,pend,find+1,iend);//创建右子树
return res;
}
}