106. 从中序与后序遍历序列构造二叉树

https://leetcode.cn/problems/construct-binary-tree-from-inorder-and-postorder-traversal/

题目要求

给定两个整数数组 inorder 和 postorder ，其中 inorder 是二叉树的中序遍历， postorder 是同一棵树的后序遍历，请你构造并返回这颗 二叉树 。

递归

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        if (postorder.length == 0)
            return null;
        TreeNode root = dfs(0, inorder.length, 0, postorder.length - 1, inorder, postorder);
        return root;
    }

    public TreeNode dfs(int inL, int inR, int postL, int postR, int[] inorder, int[] postorder) {
        // 后序数组为0，空节点
        if (inL > inR || postL > postR) {
            return null;
        }
        // 后序的最后一个元素为节点元素
        TreeNode node = new TreeNode(postorder[postR]);
        // 寻找中序数组位置切割点
        int index = inL;
        while (inorder[index] != postorder[postR]) {
            index++;
        }
        int left = index - inL;
        // 切中序数组
        node.left = dfs(inL, index - 1, postL, postL + left - 1, inorder, postorder);
        // 切后序数组
        node.right = dfs(index + 1, inR, postL + left, postR - 1, inorder, postorder);
        // 递归处理左、右区间
        return node;
    }
}

思路

• 后序遍历的最后一个元素是根节点，中序遍历根据根节点可以分割成两个区间，对应根节点的左子树和右子树；根据左子树的长度，又可以将后序遍历分割成左右两个区间；
  • 因为后序遍历是左右中的顺序，所以后序遍历前面的结点一定全是左子树的结点；