中序和后序遍历构造二叉树

Posted 2022-10-05 zcxhaha

题目链接：

涉及知识：二叉树的遍历

分析：

上一篇中介绍了如何通过二叉树的前序和中序遍历构造二叉树。

我们知道前序的遍历顺序是：根，左，右；中序的遍历顺序是左，根，右；后序的遍历顺序是左，右，根；

如果我们将后序遍历倒过来看便是根，右，左；会发现和前序遍历是非常相似的。前序遍历依次是根节点，左子树根节点，右子树根节点；后序遍历倒过来依次是根节点，右子树根节点，左子树根节点；因此解法和前序+中序生成后序的思路一样，从后序遍历中倒过来查找根节点，且先生成该节点的右子树，在生成该节点的左子树即可。

代码：

import java.util.HashMap;

/*
 * @lc app=leetcode.cn id=106 lang=java
 *
 * [106] 从中序与后序遍历序列构造二叉树
 */
/**
 * Definition for a binary tree node.
 * public class TreeNode 
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x)  val = x; 
 * 
 */
class Solution 
    private int[] inorder;
    private int[] postorder;
    private HashMap<Integer, Integer> map = new HashMap<>();
    private int tag;

    public TreeNode buildTree(int[] inorder, int[] postorder) 
        this.inorder = inorder;
        this.postorder = postorder;
        this.tag = postorder.length - 1;

        for(int i = 0; i < inorder.length; ++i)
            map.put(inorder[i], i);
        

        return generateTree(0, postorder.length - 1);
    

    //通过中序遍历的 s 到 e 生成二叉树，并返回根节点
    public TreeNode generateTree(int s, int e)
        if(s > e)
            return null;
        

        int index = map.get(postorder[tag]);
        //构造根节点
        TreeNode node = new TreeNode(postorder[tag--]);
        //生成右子树
        node.right = generateTree(index + 1, e);
        //生成左子树
        node.left = generateTree(s, index - 1);

        return node;