105. Construct Binary Tree from Preorder and Inorder Traversal

最新推荐文章于 2021-05-26 18:38:00 发布

CarryPotMan

最新推荐文章于 2021-05-26 18:38:00 发布

阅读量246

点赞数

CC 4.0 BY-SA版权

分类专栏：算法 leetcode

本文链接：https://blog.youkuaiyun.com/u012891472/article/details/50836498

算法同时被 2 个专栏收录

73 篇文章

订阅专栏

leetcode

67 篇文章

订阅专栏

本文介绍了一种基于前序遍历和中序遍历构建二叉树的方法。通过递归方式，利用前序遍历的第一个元素作为根节点，在中序遍历中找到该元素的位置，并以此划分左右子树。

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

Note:
You may assume that duplicates do not exist in the tree.

class Solution {

public:
    /* from Preorder and Inorder Traversal */
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {

        return helper(preorder,0,preorder.size(),inorder,0,inorder.size());

    }

    TreeNode* helper(vector<int>& preorder,int i,int j,vector<int>& inorder,int ii,int jj)
    {
        // tree        8 4 5 3 7 3
        // preorder    8 [4 3 3 7] [5]
        // inorder     [3 3 4 7] 8 [5]

        // 每次从 preorder 头部取一个值 mid，作为树的根节点
        // 检查 mid 在 inorder 中 的位置，则 mid 前面部分将作为 树的左子树，右部分作为树的右子树

        if(i >= j || ii >= j)
            return NULL;

        int mid = preorder[i];
        auto f = find(inorder.begin() + ii,inorder.begin() + jj,mid);

        int dis = f - inorder.begin() - ii;

        TreeNode* root = new TreeNode(mid);
        root -> left = helper(preorder,i + 1,i + 1 + dis,inorder,ii,ii + dis);
        root -> right = helper(preorder,i + 1 + dis,j,inorder,ii + dis + 1,jj);
        return root;
    }
};