DFS Construct Binary Tree from Preorder and Inorder Traversal

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

原创最新推荐文章于 2024-01-17 14:50:43 发布 · 340 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

234 篇文章

订阅专栏

本文介绍如何利用前序遍历和中序遍历来构建二叉树，详细解释了算法过程并提供了一个C++类实现。包括递归方法构建树的过程和关键步骤。

思路：

DFS。

Example：

Pre-Order：4 2 1 3 5 6

In-Order：1 2 3 4 5 6

先从Pre-Order中找到的第一个肯定是root，这里是4，然后从In-Order中以4为界，左边是root的左子树的元素，右边是root的右子树的元素，这里1，2，3是root的左子树的元素，5，6是root的右子树的元素，依次类推，再构造root的左子树和右子树，并且让root->left = 构造的左子树rootLeft & root->right = 构造的右子树的rootRight 。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    //Construct Binary Tree from Preorder and Inorder Traversal
    template<typename Iter>
    TreeNode* build(Iter pBegin, Iter pEnd, Iter iBegin, Iter iEnd) {
        if(pBegin == pEnd) return NULL;
        if(iBegin == iEnd) return NULL;
        int val = *pBegin;
        auto iRoot = find(iBegin, iEnd, val);
        TreeNode *root = new TreeNode(*iRoot);
        int leftSize = iRoot - iBegin;
        root->left = build(pBegin+1, pBegin+leftSize+1, iBegin, iRoot);
        root->right = build(pBegin+leftSize+1, pEnd, iRoot+1, iEnd);
        return root;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        size_t size = inorder.size();
        if(size == 0) return NULL;
        return build(preorder.begin(), preorder.end(), inorder.begin(), inorder.end());
    }
};