测试用例:
先序遍历{1,2,4,7,3,5,6,8},中序遍历{4,7,2,1,5,3,8,6};
输出结果:
重构二叉树的先序遍历。
编程思路:
我们观察先序遍历和中序遍历的结果。先序遍历的第一个节点{1}肯定是根节点,然后在中序遍历中找到此根节点{1},我们可以看到,中序遍历中,根节点以左是二叉树的左子树{4,7,2},根节点以右是二叉树的右子树{5,3,8,6}。然后,左右子女树又可以单独的看成一个独立的二叉树,然后再对其划分,如此递归,便可重构二叉树的结构。
代码:
import java.util.Arrays;
public class Test{
public static void main(String args[]){
Solution s = new Solution();
int[] pre={1,2,4,7,3,5,6,8};
int[] in={4,7,2,1,5,3,8,6};
TreeNode result = s.reConstructBinaryTree(pre, in);
s.showBinTree(result);
}
}
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { this.val = x; }
}
class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre==null){
return null;
}
TreeNode root=new TreeNode(0);
int i=0;
int[] lpre=null,lin=null,rpre=null,rin=null;
root.val=pre[0];
while(in[i]!=pre[0]){
i++;
}
if(i>0){
lpre=Arrays.copyOfRange(pre,1,i+1);
lin=Arrays.copyOfRange(in,0,i);
}
if(i<pre.length-1){
rpre=Arrays.copyOfRange(pre,i+1,pre.length);
rin=Arrays.copyOfRange(in,i+1,in.length);
}
root.left=reConstructBinaryTree(lpre,lin);
root.right=reConstructBinaryTree(rpre,rin);
return root;
}
public void showBinTree(TreeNode n){
if(n!=null){
System.out.println(n.val);
}
else return;
showBinTree(n.left);
showBinTree(n.right);
}
}