LeetCode 106. 已知中序&后序求二叉树

最新推荐文章于 2024-06-27 19:32:22 发布

Michael阿明

最新推荐文章于 2024-06-27 19:32:22 发布

阅读量1.2k

点赞数 1

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_21201267/article/details/100753691

版权

LeetCode 专栏收录该内容

1849 篇文章 126 订阅

订阅专栏

文章目录

1. 题目

在这里插入图片描述
类似题目LeetCode 105. 已知前序&中序求二叉树

2. 解题

2.1 递归

递归法，后序最后一个是根节点
在这里插入图片描述

class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        unordered_map<int,int> m;//哈希表
        for(int i = 0; i < inorder.size(); ++i) 
        {
            m[inorder[i]] = i;//方便后面查找位置
        }
        return build(postorder, inorder,0,postorder.size()-1,0,inorder.size()-1,m);
    }
    TreeNode* build(vector<int>& postorder, vector<int>& inorder, int pS, int pE, int iS, int iE, unordered_map<int,int> &m)
    {
        if(pS > pE)
            return NULL;
        TreeNode *root = new TreeNode(postorder[pE]);
        int leftlen = m[postorder[pE]]-iS;
        root->left = build(postorder,inorder,pS,pS+leftlen-1,iS,m[postorder[pE]]-1,m);
        root->right = build(postorder,inorder,pS+leftlen,pE-1,m[postorder[pE]]+1,iE,m);
        return root;
    }
};

2.2 循环

class Solution {    //别人写的 循环
public:
    TreeNode* buildTree(vector<int>& in, vector<int>& post) 
    {
        if (in.empty()) 
            return NULL;
        stack<TreeNode*> S;
        TreeNode *root = new TreeNode(post.back()), *cur = root;
        S.push(root);
        for (int i = post.size() - 2, j = in.size() - 1; i >= 0; i--) 
        {
            TreeNode *back = NULL, *cur = new TreeNode(post[i]);
            while (!S.empty() && S.top()->val == in[j]) 
            {
                back = S.top(), 
                S.pop(), 
                j--;
            }
            if (back)
                back->left = cur;
            else
                S.top()->right = cur;
            S.push(cur);
        }
        return root;
    }
};