Java解决重建二叉树算法题

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

解题思路

由二叉树的前序遍历序列可知,1 为二叉树的根节点。由二叉树的中序遍历序列可知,1之前的结点,即4、7、2为其左子树的中序遍历序列,1之后的结点 5、3、8、6为其右子树的中序遍历序列。因此,左子树共有三个节点,右子树有4个结点。所以,在前序遍历序列中,1结点后的三个结点就是左子树三个结点的值,最后的四个结点就是右子树的四个结点的值。
这样,我们就分别找到了二叉树的根节点,左、右子树的前序和中序遍历序列,因此可以使用递归的方法去完成问题。
Java解决重建二叉树算法题

JAVA代码

/**
 * 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) throws Exception{
        if(pre==null||in==null||pre.length!=in.length){
            return null;
        }
        return ConstructCore(pre,0,pre.length-1,in,0,in.length-1);
    }
    
    public TreeNode ConstructCore(int[] pre,int preStart,int preEnd,
                            int[] in,int inStart,int inEnd) throws Exception{
        int rootVal = pre[preStart];
        TreeNode root = new TreeNode(rootVal);
        if(preStart == preEnd){
            if(inStart == inEnd && pre[preStart] == in[inStart]){
                return root;
            }else{
                throw new Exception("Invalid input data");
            }
        }
        //在中序遍历中找到根节点的值
        int inRoot = inStart;
        while(inRoot < inEnd && in[inRoot] != rootVal){
            inRoot++;
        }
        if(inRoot==inEnd && in[inRoot]!=rootVal){
            throw new Exception("Invalid input data");
        }
        int leftLength = inRoot - inStart;
        int rightLength = inEnd - inRoot;
        if(leftLength>0){
            root.left = ConstructCore(pre,preStart+1,preStart+leftLength,in,inStart,inRoot-1);
        }
        if(rightLength>0){
            root.right = ConstructCore(pre,preStart+leftLength+1,preEnd,in,inRoot+1,inEnd);
        }
        return root;
    }
}
上一篇:[19/09/16-星期一] Python的运算符和条件判断语句


下一篇:最低公共祖先(LCA)