106. Construct Binary Tree from Inorder and Postorder Traversal 根据中序、后序遍历序列构建二叉树(Java)

题目:

Given inorder and postorder traversal of a tree, construct the binary tree.

Note: You may assume that duplicates do not exist in the tree.
106. Construct Binary Tree from Inorder and Postorder Traversal 根据中序、后序遍历序列构建二叉树(Java)_第1张图片

解答:

本题根据树的中序遍历和后序遍历,得到树的结构。与 LeetCode 105 思路基本一致

整体思路为:
后序遍历的最后一个节点一定是根节点,根据后序遍历确定根节点后,在中序中找到根节点的位置,则该位置之前的均为根节点的左子树,该位置之后的都是根节点的右子树,以此进行递归实现

具体代码实现如下:

  1. 用 post 指向后序遍历中,当前的根节点,初始指向后序遍历的最后一个节点
  2. 用 index 指向中序遍历中的根节点,则 index 左侧的为 root 的左子树,index 右侧的为 root 的右子树
  3. 递归中,inStart 和 inEnd 分别指向当前左子树/右子树的节点范围,分别递归,找到 root 的左子树和右子树
  4. 其中递归实现时,root 的右子树根节点位置指向后序遍历 postorder 中的 post-1,root 的左子树根节点位置指向后序遍历 postorder 中的 post-(inEnd-index+1),其中 inEnd-index+1 为根节点右子树的节点个数,则根节点的左子树节点指向 post 向前移动右子树节点的个数个位置
class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        if(inorder==null || postorder==null) {
            return null;
        }
        return build(inorder, postorder, 0, inorder.length-1, postorder.length-1);
    }
    private TreeNode build(int[] inorder, int[] postorder, int inStart, int inEnd, int post) {
        if(inStart > inEnd || post <0) {
            return null;
        }
        TreeNode root = new TreeNode(postorder[post]);
        int index = 0;
        for(int i=inStart; i<=inEnd; i++) {
            if(inorder[i] == root.val) {
                index = i;
                break;
            }
        }
        root.right = build(inorder, postorder, index+1, inEnd, post-1);
        root.left = build(inorder, postorder, inStart, index-1, post-(inEnd-index+1));
        return root;
    }
}

你可能感兴趣的:(LeetCode)