根据一棵树的前序遍历与中序遍历构造二叉树。
思路:
有了前序遍历,首先我们可以知道根节点的值,也就是数组中下标为0的位置,由此创建根节点。
然后在中序遍历中找到根的值所在的下标,切出左右子树的前序和中序。
注意:如果前序遍历的数组长度为0,说明是一棵空树。
下面用具体的例子来说明:
前序:ABCDE
中序:CDBAE
1、首先可以确定A是这棵树的根节点,然后根据中序遍历切出A的左右子树,得到BCD属于A的左子树,E属于A的右子树,如下图:
2、接着以B为根节点,在中序遍历中再次切出B的左右子树,得到CD为B的左子树,右子树为空。
3、再以C为根节点,结合中序遍历,得到D为C的右子树,左子树为空。
创建好的二叉树如下图所示:
具体实现代码:
public class CreateBinaryTreeQZ {
public static class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
private static int find(char[] array, int v){
for (int i=0;i<array.length;i++){
if (array[i] == v){
return i;
}
}
return -1;
}
private static TreeNode buildTree(char[] preorder, char[] inorder) {
if (preorder.length==0){
return null;
}
//1、根据前序,找到根的值,并且创建根节点
char rootValue = preorder[0];
TreeNode root = new TreeNode(rootValue);
//2、在中序中找到根的值所在的下标
int leftSize = find(inorder,rootValue);
//3、切出左子树的前序和中序
char[] leftPreorder = Arrays.copyOfRange(preorder,1,1+leftSize);
char[] leftInorder = Arrays.copyOfRange(inorder,0,leftSize);
root.left = buildTree(leftPreorder,leftInorder);
//4、切出右子树的前序和中序
char[] rightPreorder = Arrays.copyOfRange(preorder,1+leftSize,preorder.length);
char[] rightInorder = Arrays.copyOfRange(inorder,leftSize+1,preorder.length);
root.right = buildTree(rightPreorder,rightInorder);
return root;
}
public static void main(String[] args) {
char[] preOrder = new char[]{'A','B','C','D','E'};
char[] inorder = new char[]{'C','D','B','A','E'};
TreeNode root = buildTree(preOrder,inorder);
System.out.println("创建成功");
}
}