[Lintcode]Construct Binary Tree from Inorder and Postorder Traversal

最新推荐文章于 2021-02-03 15:56:42 发布

青铁

最新推荐文章于 2021-02-03 15:56:42 发布

阅读量180

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： lintcode

本文链接：https://blog.youkuaiyun.com/jc69186918/article/details/52431302

算法专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种使用递归算法根据给定的中序和后序遍历构建二叉树的方法。通过确定根节点并划分左右子树来实现树的重建。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

Example

Given inorder [1,2,3] and postorder [1,3,2], return a tree:

  2
 / \
1   3

递归算法，从后序数组确定每次遍历的根节点，再从中序数组中找到根节点，即可确定根节点左右子树。参数需要中序数组开始/结束位置以及后续数组当前根节点的位置。

public class Solution {
    /**
     *@param inorder : A list of integers that inorder traversal of a tree
     *@param postorder : A list of integers that postorder traversal of a tree
     *@return : Root of a tree
     */
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        return helper(inorder, postorder, 0, inorder.length -1, inorder.length -1);
    }
    
    //1) need endP for postorder
    TreeNode helper(int[] inorder, int[] postorder, int start, int end, int endP) {
        if(start > end) return null;
        if(start == end) return new TreeNode(inorder[start]);
        int root = postorder[endP];
        int rootIndex = -1;
        for(int i = start; i <= end; i++) {
            if(inorder[i] == root) {
                rootIndex = i;
            }
        }
        TreeNode res = new TreeNode(root);//2) calculate carefully the postorder index
        res.left = helper(inorder, postorder, start, rootIndex - 1, endP - (end - rootIndex) - 1);
        res.right = helper(inorder, postorder, rootIndex + 1, end, endP - 1);
        return res;
    }
}