通用方式解决根据二叉树遍历重建二叉树问题

通用方式解决根据二叉树遍历重建二叉树问题

重做LeetCode题目的时候，发现两道题目非常类似，但之前却用了两种写法差别比较大的解法。感觉可以合并为同一种方式进行解决，于是找出了如下的通用方式解决这两个问题

题目一：

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如，给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

题目二：

根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如，给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]

可以看出仅仅是给出的两种遍历方式的不同。
大家都知道根据 前序+中序遍历 或 中序+后序遍历可以唯一确定一棵二叉树
前序遍历：根 左 右
中序遍历：左 根 右
后序遍历：左 右 根
这两道题的思路都非常类似：
根据前序遍历或后序遍历确定出根节点
在从中序遍历中获取其左右子树
递归上述步骤即可

题目一代码：

class Solution {
    private int preorder_index;
    private int[] inorder;
    private int[] preorder;
    private Map<Integer,Integer> idx_map;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        this.inorder = inorder;
        this.preorder = preorder;
        this.preorder_index = 0;
        this.idx_map = new HashMap<>();
        for (int i = 0; i < inorder.length; i++) {
            idx_map.put(inorder[i],i);
        }
        return buildTree(0,preorder.length-1);
    }

    public TreeNode buildTree(int left,int right){
        if (left > right)
            return null;
        TreeNode root = new TreeNode(preorder[preorder_index]);
        Integer in_index = idx_map.get(root.val);
        this.preorder_index++;
        root.left = buildTree(left,in_index-1);
        root.right = buildTree(in_index+1,right);
        return root;
    }
}

题目二代码：

class Solution {
    private int postorder_index;
    private int[] inorder;
    private int[] postorder;
    private Map<Integer,Integer> idx_map;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        //初始化成员变量
        this.inorder = inorder;
        this.postorder = postorder;
        this.idx_map = new HashMap<>();
        this.postorder_index = postorder.length - 1;    //从后序遍历的最后一个元素开始
        //将中序遍历的 元素 - 下标 作为K V 放入map集合
        for (int i = 0; i < inorder.length; i++) {
            this.idx_map.put(inorder[i],i);
        }
        TreeNode root = handler(0, postorder.length - 1);
        return root;
    }

    /**
     * 构建树的处理方法
     * @param left 后序遍历的左指针
     * @param right 后序遍历的右指针
     * @return
     */
    public TreeNode handler(int left,int right){
        //当左指针 > 右指针代表已经没有结点元素 可以用来构建树 直接返回null
        if(left > right){
            return null;
        }
        //有元素的情况下 取出后序遍历的最后一个元素作为跟结点
        TreeNode node = new TreeNode(postorder[postorder_index]);
        //找到根结点在中序遍历中的位置 ， 将中序遍历结果分为左右两棵子树
        int index = idx_map.get(postorder[postorder_index]);
        //将后序遍历长度-1 取出当前的元素结点
        postorder_index --;
        node.right = handler(index+1,right);
        node.left = handler(left,index-1);
        return node;
    }
}

代码差距只有一行 找找看