剑指Offer07.重建二叉树_解题思路代码实现
创始人
2024-02-19 12:42:36
0

一、题目描述

输入某二叉树的前序遍历中序遍历的结果,请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

二、示例输入输出

示例1:
在这里插入图片描述
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]

三、题目解析 & 解题思路

思路可知输入为前序遍历、中序遍历
前序遍历:前序遍历顺序为 根左右 即 先访问根,再遍历左子树,再遍历右子树
中序遍历:中序遍历顺序为 左根右 即 先访问左子树,再访问根节点,最后访问 右子树
前序遍历顺序特点:第一个 为根节点 例如 示例 1 为: 3 9 20 15 7 可以得知 节点 3 为整棵二叉树的根节点,再结合中序遍历 的顺序 :9 3 15 20 7 可以得知 节点 3 左侧的 为整棵树的 左子树,节点 3 右边的为整棵树的 右子树
首先可以确定的就是根节点 为 前序遍历的第一个
如下图示例:

在这里插入图片描述
根据前序遍历特点:根左右 那么 前序遍历的结果中 左子树为 245 恰好 可以根据 中序遍历 中的根节点 1 划分 左子树 为 245 和 367 接下来 开始 根据前序 的 245 和 中序 的 425 来 判断 节点 2、4、5 这三个节点 那个是左孩子那个是右孩子即可

3.1

首先将 前序遍历中的第一个节点 作为根节点 插入到 二叉树中,若前序遍历为空 那么整个树也为空

3.2

另外 我们需要一个 栈进行辅助,保存前序遍历的节点(根节点或左孩子节点 因为 前序 根左右的 遍历顺序 会一直,根 -> 左(父) -> 左(父) 直到遍历到最后一个 左孩子 叶子节点,才会 回溯 回去 遍历 上一层的 右孩子节点),且 右孩子节点肯定是 栈中 某一个节点的 右孩子节点
并从下标 index = 0 判断中序遍历 下标为 index 的元素 是否等于当前栈顶元素 如果等于栈顶元素,说明该子树 左孩子或父节点 已经遍历完毕
在这里插入图片描述

例如 假如 前序遍历和中序遍历的第一个节点都是 1 那么说明,这棵树没有左子树:
在这里插入图片描述

按照这样的思路 我们先 入栈 节点 124
stack [1, 2 ,4]
并且将这些节点 作为 左孩子 逐步插入,当遍历 到 前序 1245367 中的 5 时 我们发现 现在的栈顶 元素为 4 且 中序遍历顺序的第一个 元素也为 4 ,且中序遍历为 左根右 ,因此 节点 4 没有左孩子,那么在前序遍历 1245367 节点 4 的 下一个元素 5 ,肯定是 124 某一个节点 的 右孩子节点
index 依次向后移动,并和栈顶元素比较,因为栈里都是 根 左 根 左 ,栈顶到 栈底的顺序 为 左 根 左 根 ,而中序为 左 根 右 ,当 中序 下标为 index 且 与 栈顶元素相等 时,index 继续向后移动,栈 弹出 栈顶元素,当栈顶元素不等于 index 元素 时 说明遇到了 右孩子,且该节点是 上一次弹出的 节点的右孩子
即 5 != 1 则 5 为上一个弹出 的 节点 2 的 右孩子,这时 我们将 5 作为 右孩子 插入

总体思路:在前序中 构建节点的左孩子,中序中 确定 右孩子节点

四、代码实现

/*** 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 {
public:TreeNode* buildTree(vector& preorder, vector& inorder) {if(preorder.size() == 0 || inorder.size() == 0){return nullptr;}stack stkTree;//前序遍历结果的第一个作为 根节点插入TreeNode* Root = new TreeNode(preorder[0]);stkTree.push(Root);int inorderIndex = 0;for(int i = 1; i < preorder.size(); i++){int iTemp = preorder[i];TreeNode* nodeTemp = stkTree.top();//从前序中构建出来左孩子节点 if(nodeTemp->val != inorder[inorderIndex]){TreeNode* node = new TreeNode(iTemp);nodeTemp->left = node;stkTree.push(nodeTemp->left);}else{//从后序节点中找到 右孩子节点while(!stkTree.empty() && (stkTree.top()->val == inorder[inorderIndex])){//保存节点,最后一个弹出的为右孩子的父节点nodeTemp = stkTree.top();stkTree.pop();++inorderIndex;}nodeTemp->right = new TreeNode(iTemp);stkTree.push(nodeTemp->right);}}return Root;}
};

相关内容

热门资讯

监控摄像头接入GB28181平... 流程简介将监控摄像头的视频在网站和APP中直播,要解决的几个问题是:1&...
Windows10添加群晖磁盘... 在使用群晖NAS时,我们需要通过本地映射的方式把NAS映射成本地的一块磁盘使用。 通过...
protocol buffer... 目录 目录 什么是protocol buffer 1.protobuf 1.1安装  1.2使用...
在Word、WPS中插入AxM... 引言 我最近需要写一些文章,在排版时发现AxMath插入的公式竟然会导致行间距异常&#...
Fluent中创建监测点 1 概述某些仿真问题,需要创建监测点,用于获取空间定点的数据࿰...
educoder数据结构与算法...                                                   ...
MySQL下载和安装(Wind... 前言:刚换了一台电脑,里面所有东西都需要重新配置,习惯了所...
MFC文件操作  MFC提供了一个文件操作的基类CFile,这个类提供了一个没有缓存的二进制格式的磁盘...
有效的括号 一、题目 给定一个只包括 '(',')','{','}'...
【Ctfer训练计划】——(三... 作者名:Demo不是emo  主页面链接:主页传送门 创作初心ÿ...