DFS Construct Binary Tree from Inorder and Postorder Traversal

思路:

DFS,与上一题思考方式一样,这次的root出现在Post-Order的末尾。

Example:


In-Order: 1 2 3 4 5 6

Post-Order: 1 3 2 6 5 4

先从Post-Order末尾找到root,这里是4,然后在In-Order中找到root的位置,左半部分就是root的左子树的元素,右半部分就是root右子树的所有元素,这里1,2,3是root的左子树的元素,5,6是root右子树的元素,依次类推,再构造root的左子树和右子树,并且让root->left = 构造的左子树 & root->right = 构造的右子树。

/**
 * 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 Inorder and Postorder Traversal
    template<typename Iter>
    TreeNode* build(Iter iBegin, Iter iEnd, Iter pBegin, Iter pEnd) {
        if(iBegin == iEnd) return NULL;
        if(pBegin == pEnd) return NULL;
        int val = *prev(pEnd);
        auto iRoot = find(iBegin, iEnd, val);
        TreeNode *root = new TreeNode(*iRoot);
        int leftSize = iRoot - iBegin;
        root->left = build(iBegin, iRoot, pBegin, pBegin + leftSize);
        root->right = build(iRoot + 1, iEnd, pBegin + leftSize, prev(pEnd));
        return root;
    }
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        size_t size = postorder.size();
        if(size == 0) return NULL;
        return build(inorder.begin(), inorder.end(), postorder.begin(), postorder.end());
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值