题解 | #重建二叉树#

重建二叉树

http://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6

前提知识:
前序遍历 : 节点按照[ 根 | 左 | 右 ] 顺序遍历
中序遍历: 节点按照[ 左 | 根 |右 ] 顺序遍历
由此可知:
1.前序遍历的首元素为树的根节点值。
2.根节点的值可以将中序遍历划分为 [左子树中序遍历 | 根节点 | 右子树中序遍历 ]。
3.根据中序遍历左右子树的节点个数,可以将前序遍历划分为 [根节点 | 左子树前序遍历 | 右子树前序遍历 ]
图片说明

题解一:递归
题解思路: 通过上述分析,可以递归的处理所有子树
递归分析:
1.递归参数: pre前序遍历数组, vin 中序遍历数组, left中序遍历左边界,right中序遍历右边界
2. 递归边界: left>right 返回NULL
3.回溯返回 root
并且为了降低查询根节点在中序中索引的时间复杂度,我们利用哈希表快速定位。
图示:
图片说明

复杂度分析:
时间复杂度:O(N),N节点个数
空间复杂度:O(N),使用O(N)空间存储哈希映射。其次,递归栈深度最优O(logN)此时二叉树为满二叉树;最坏O(N):树退化为链表

实现如下:

class Solution {
public:
    unordered_map<int, int> mp;
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        for(int i = 0;i<vin.size();i++) mp[vin[i]] = i;
        return recovery(pre, vin, 0, vin.size()-1);
    }
    TreeNode* recovery(vector<int> &pre,vector<int> vin,int left,int right){
        if(right<left) return NULL;// 递归终止
        TreeNode* root = new TreeNode(pre[0]);// 建立根节点
        pre.erase(pre.begin());// 删除前序的首元素
        int mid = mp[root->val];// 根节点在中序遍历的索引值
        root->left = recovery(pre, vin, left, mid-1);//递归左子树
        root->right = recovery(pre,vin,mid+1,right);//递归右子树
        return root;
    }
};

前提: 前序遍历相邻两个值a,b之间的关系
1:a有左子树,那么b就是a的直接左儿子;
2: a没有左子树但是有右子树,b为a的右儿子; a为叶子节点,那么b为a祖先节点的右儿子;
图示:
情况1:
图片说明
情况2:
图片说明

题解二:迭代(利用辅助栈)
题解思路:
1.根据上述分析的关系,我们使用栈保存祖先节点和指针inorder来遍历中序遍历数组。
2. 我们一次枚举前序遍历中除了第一个节点外的每个节点。 如果indorder等于栈顶元素,那么就不断地弹出栈顶元素并移动inorder, 并将当前节点作为最后一个弹出的节点的右儿子;如果不同,那么就将当前节点作为栈顶节点的左儿子;
复杂度分析:
时间复杂度:O(N)
空间复杂度:O(N)

实现如下:

class Solution {
public:
     TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.size()==0) return NULL; //判空
        stack<TreeNode*> st;
        TreeNode* root = new TreeNode(pre[0]); //创建根节点
        st.push(root); //根节点入栈
        int inorder = 0;
        //遍历前序遍历
        for(int i = 1;i<pre.size();i++)
        {
            TreeNode * node = st.top(); //获取栈顶
            if(node->val != vin[inorder])
            {
                //如果不同,那么就将当前节点作为栈顶节点的左儿子
                node->left = new TreeNode(pre[i]);
                st.push(node->left);
            }
            else{
                //如果相等,则前序遍历中的左子树所有节点已经入栈
                //根据中序遍历,对左子树上节点的重建
                while(!st.empty() && st.top()->val == vin[inorder])
                {
                    node = st.top();
                    st.pop();
                    inorder++;
                }
                //当前节点作为最后一个弹出的节点的右儿子
                node->right = new TreeNode(pre[i]);
                st.push(node->right);
            }
        }
    return root;
    }
};

此题同JZ4

牛客网编程题题解 文章被收录于专栏

本专栏记录在牛客网上AC的每一题,写下题解。 未来2年完成2000编程题的题解。 2021.12.29更新,最进准备毕设,断更了,会尽快做完毕设,继续做这一件事情

全部评论
你这个方法做出来有问题
点赞 回复 分享
发布于 2021-08-30 09:46

相关推荐

沉淀一会:**圣经 1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
小红书 后端开发 总包n+8w+期权
点赞 评论 收藏
分享
评论
5
收藏
分享
正在热议
# 25届秋招总结 #
443331次浏览 4520人参与
# 春招别灰心,我们一人来一句鼓励 #
42187次浏览 537人参与
# 北方华创开奖 #
107473次浏览 600人参与
# 地方国企笔面经互助 #
7973次浏览 18人参与
# 同bg的你秋招战况如何? #
77166次浏览 569人参与
# 实习必须要去大厂吗? #
55811次浏览 961人参与
# 阿里云管培生offer #
120451次浏览 2220人参与
# 虾皮求职进展汇总 #
116310次浏览 887人参与
# 如果你有一天可以担任公司的CEO,你会做哪三件事? #
11683次浏览 289人参与
# 实习,投递多份简历没人回复怎么办 #
2454962次浏览 34861人参与
# 提前批简历挂麻了怎么办 #
149927次浏览 1978人参与
# 在找工作求抱抱 #
906124次浏览 9423人参与
# 如果公司给你放一天假,你会怎么度过? #
4762次浏览 55人参与
# 你投递的公司有几家约面了? #
33209次浏览 188人参与
# 投递实习岗位前的准备 #
1196037次浏览 18550人参与
# 机械人春招想让哪家公司来捞你? #
157648次浏览 2267人参与
# 双非本科求职如何逆袭 #
662384次浏览 7397人参与
# 发工资后,你做的第一件事是什么 #
12806次浏览 62人参与
# 工作中,努力重要还是选择重要? #
35906次浏览 384人参与
# 简历中的项目经历要怎么写? #
86937次浏览 1516人参与
# 参加完秋招的机械人,还参加春招吗? #
20153次浏览 240人参与
# 我的上岸简历长这样 #
452074次浏览 8089人参与
牛客网
牛客企业服务