剑指第二版第7题重建二叉树

剑指第二版第7题重建二叉树

这算是一道老题目了,通过前续遍历和中序遍历创建一个二叉树,我记得还有一道反序列二叉树的题,只给了三种dfs中的一种,区别还是很大的,没有什么可以说的了,找规律就可以了.

class Solution {
      HashMap<Integer, Integer> map = new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        //把中序遍历的数进行存储
        if (preorder.length == 0) {
            return null;
        }
        if (preorder.length == 1 || inorder.length == 1) {
            return new TreeNode(preorder[0]);
        }
        for (int i = 0; i < inorder.length; i++) {
            map.put(inorder[i],i);
        }
        return buildTreeNode(preorder,inorder,0,preorder.length - 1,0,inorder.length - 1);
    }

    //在合理控制数组的下标,对二叉树进行重新的构建
    public TreeNode buildTreeNode(int[] preorder, int[] inorder, int ps, int pe, int is, int ie) {
        if (ps > pe) {
            return null;
        }
        Integer index= map.get(preorder[ps]);
        //这里还是需要构造一个左子树的变量
        int leftSize = index - is;
        TreeNode treeNode = new TreeNode(preorder[ps]);
        treeNode.left = buildTreeNode(preorder, inorder,ps + 1,ps + leftSize,is,index - 1);
        treeNode.right = buildTreeNode( preorder,inorder,ps + leftSize + 1,pe,index + 1,ie);
        
        return treeNode;
    }
}