JAVA通过前序,中序重建二叉树
在leetcode,剑指offer中遇到重建二叉树这个题
给定节点数为 n 二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。
输入: [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) 根据前序遍历的遍历算法:根节点-左子树-右子树,可以得出第一个元素必然为根节点;
(2) 既然找到了根节点,那么再将这个根节点和中序遍历中的子节点对比一下,可以得出,中序遍历中根节点的左边元素为二叉树的左子树,右边元素为二叉树的右子节点;
举一个例子:
在前序遍历[1,2,4,7,3,5,6,8]中1为根节点,那么将1在[4,7,2,1,5,3,8,6]中进行遍历,得出[4,7,2]为根节点的左子树,而[5,3,8,6]为根节点的右子树。
(3) 根据解题思路1我们可以轻松的得出根节点为1,那么在前序遍历中如何确定他的左子树的个数呢?对,我们解题思路2,中序遍历发现,左子树为三个元素。即在前序遍历[1,2,4,7,3,5,6,8]中[1]为根节点,[2,4,7]为左子树,【3,5,6,8】为右子树;
(4) 我们再来思考一步,在[2,4,7]中能确定什么?[2]为根节点,[4,7] 能确定是左子节点还是右子节点吗?——不能,那我们来看一下中序遍历[4,7,2]这个左子树中,根节点[2]的左边为左子树,由此发现了递归的规律,当前序遍历为[4,7]时,中序遍历为[4,7],根节点为4时,在中序遍历7在4的右边,为右子节点。于是整个思路得到了验证。如图所示。
三、how?
我们需要找到递归条件:
1.当前序遍历的根节点==中序遍历的元素时(所以这里需要用for循环将中序遍历的元素遍历出来。)
2. 用for循环时遍历出来的i别闲着,用起来,我们需要它作为遍历数组下标;
四、上代码:
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre.length == 0){
System.out.println("数组不能为空!")
return null;
}
int rootVal = pre[0];
if(pre.length == 1){
return new TreeNode(rootVal);
}
TreeNode root = new TreeNode(rootVal);
int rootIndex = 0;
for(int i=0;i<in.length;i++){
if(rootVal == in[i]){
rootIndex = i;
break;
}
}
root.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,rootIndex+1),Arrays.copyOfRange(in,0,rootIndex));
root.right = reConstructBinaryTree(Arrays.copyOfRange(pre,rootIndex+1,pre.length),Arrays.copyOfRange(in,rootIndex+1,in.length));
return root;
}
}