题解 | 从中序与后序遍历序列构造二叉树 | 简单递归

从中序与后序遍历序列构造二叉树

https://www.nowcoder.com/practice/ab8dde7f01f3440fbbb7993d2411a46b

解题思路

简单使用递归即可。

1、确定根节点,必然是后序遍历的最后一个元素,以此创建树。

2、找到根节点在中序遍历中的位置。

3、递归创建左右子树。

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <algorithm>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param inorder int整型vector 中序遍历序列
     * @param postorder int整型vector 后序遍历序列
     * @return TreeNode类
     */
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if(inorder.empty()) return nullptr; //返回空树
        int root = postorder.back();    //后序遍历的最后一个元素必为根节点
        auto tree = new TreeNode(root); //以根节点创建新树
        postorder.pop_back();           //后序遍历集合中去除根节点
        int pos = find(inorder.begin(), inorder.end(), root) - inorder.begin(); //获取根节点在中序遍历中的位置
        vector<int> inLeft(inorder.begin(), inorder.begin() + pos);
        vector<int> inRight(inorder.begin() + pos + 1, inorder.end());
        vector<int> postLeft(postorder.begin(), postorder.begin() + pos);
        vector<int> postRight(postorder.begin() + pos, postorder.end());
        tree->left = buildTree(inLeft, postLeft);
        tree->right = buildTree(inRight, postRight);
        return tree;
    }
};

全部评论

相关推荐

09-01 16:46
已编辑
门头沟学院 Java
mmvvpp:错了!!给了offer之后还有试用期,试用期过了就完事了?错了!还有每个季度的kpi考核,拿一个c就等着被劝退。那我好好干不拿c不就完了?错了!最多三年劳动合同到期,续不续期未知数。每年都有1800w毕业生毕业,今年你是小萌新蜜月期,明年你是老油条,长江后浪推前浪,前浪死在沙滩上。这就是——互联网!
秋招的破防瞬间
点赞 评论 收藏
分享
渴望wlb的牛油果很...:直说卡第一学历不就行了 非得拐弯抹角
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务