题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
对于给的样例,树的结构应该为
1
/ \
2 3
/ / \
4 5 6
\ /
7 8
先序遍历和中序遍历的规律为:先序遍历的第一个节点为根( 1),中序遍历中找到该节点(即 1),在中序遍历中,在1的左边的为1 的左子树,在2 的右边的为右子树。
在左子树中找(4,7,2)。然后在前序遍历的再往后找到2(后面的第一个),在中序遍历中找到2,在2的左边的(4,7)为左子树,右边的( 空)为右子树。
在左子树中找(4,7)。然后在前序遍历中再往后找到4(后面的第一个),在中序遍历中找到4,在4的左边(空)的为左子树,右边的(7)为右子树。
在左子树中找(7)。然后在前序遍历中再往后找到7(后面的第一个),在中序遍历中找到7,在4的左边(空)的为左子树,右边的(空)为右子树。
结束。
到右子树,和左边相似。
递归的代码为:
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 = 0; i <=preend ; i++) {
if (in[i] == pre[prestart]) {
root.left = reConstructBinaryTree(pre,prestart+1,i-instart+prestart,in,instart,i);
root.right = reConstructBinaryTree(pre,i-instart+prestart+1,preend,in,i+1,inend);
}
}
return root;
}
}