*[Lintcode]Construct Binary Tree from Preorder and Inorder Traversal

最新推荐文章于 2024-01-17 14:50:43 发布

青铁

最新推荐文章于 2024-01-17 14:50:43 发布

阅读量193

点赞数

CC 4.0 BY-SA版权

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

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

算法专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种通过前序遍历和中序遍历构建二叉树的方法。利用递归方式定位根节点，并以此构建左右子树，最终形成完整的二叉树。

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

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

Example

Given in-order [1,2,3] and pre-order [2,1,3], return a tree:

  2
 / \
1   3

以中序数列为基础，记录中序数列的start，end以及前序数列的start。对于一个中序数列的给定区间，从前序数列寻找根节点。需要注意的是每次递归时起点是start而不是0

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