力扣刷题 二叉树的迭代遍历

题干

给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

示例 1:

输入:root = [1,null,2,3]
输出:[1,2,3]

示例 2:

输入:root = []
输出:[]

示例 3:

输入:root = [1]
输出:[1]

示例 4:

输入:root = [1,2]
输出:[1,2]

示例 5:

输入:root = [1,null,2]
输出:[1,2]

解题思路

我们上次使用了递归来完成前中后序遍历,这次我们用迭代法来实现。

前序遍历

大体思路是用栈来实现遍历。我们先建立一个栈,把根节点root压入栈。

我们模拟从根节点root开始,先设置一个循环,当栈不为空的时候,对每一个栈顶元素循环操作。因此根节点只是一个个例,后面在写代码的时候不能只考虑根节点的情形。

我们先把root从栈中弹出,如果它不为空,则将其放入结果数组中。如果为空的话,就跳出本次循环,对下一个栈元素进行操作。(根节点是第一个栈元素,自然不会有下一个栈元素,但是我们写代码考虑的是全局情况)

然后依次把它的右子树和左子树压入栈中,因为栈是后进先出的,所以下一个出栈的就是左子树,符合中左右的顺序。

//迭代法
class Solution {
public:vector<int> preorderTraversal(TreeNode* root) {//建立一个栈stack<TreeNode*> st;//建立一个数组保存遍历结果vector<int> result;//将根节点入栈st.push(root); // 在栈不为空的情况下,重复以下操作while(!st.empty()){//将栈顶元素弹出TreeNode* node = st.top();st.pop();//当根节点不为空if(node != NULL){// 将栈顶元素放入结果数组result.push_back(node->val);}//若为空, 则忽略这个节点,去访问下一个节点else{continue;}//将节点的右子树的根节点放入栈st.push(node->right);//将节点的左子树的根节点放入栈,这样下一个遍历的就是左子树st.push(node->left); }// 返回遍历的结果return result;}
};

写完了前序遍历的迭代法遍历,后序遍历就可以轻松写出了,因为前序遍历的顺序是中左右,而后序遍历是左右中。想要从前序变为后序,只需先将左右调换位置,变成中右左,再翻转数组就实现左右中了。而左右调换位置和简单,只需将入栈的顺序调换即可。而翻转函数之前我们也接触过。

后序遍历代码如下

这里我稍微修改了写法,把根节点root和普通根节点的情况做了区分,可能更容易理解,但本质相同。

class Solution {
public:vector<int> postorderTraversal(TreeNode* root) {vector<int> result;stack<TreeNode*> st;if (root != NULL)st.push(root);while (!st.empty()) {TreeNode* node = st.top();st.pop();result.push_back(node->val);if (node->left)st.push(node->left);if (node->right)st.push(node->right);}reverse(result.begin(), result.end());return result;}
};

而中序遍历就没有那么简单了。

为了解释清楚,我说明一下 刚刚在迭代的过程中,其实我们有两个操作:

  1. 处理:将元素放进result数组中
  2. 访问:遍历节点

分析一下为什么刚刚写的前序遍历的代码,不能和中序遍历通用呢,因为前序遍历的顺序是中左右,先访问的元素是中间节点,要处理的元素也是中间节点,所以刚刚才能写出相对简洁的代码,因为要访问的元素和要处理的元素顺序是一致的,都是中间节点。

那么再看看中序遍历,中序遍历是左中右,先访问的是二叉树顶部的节点,然后一层一层向下访问,直到到达树左面的最底部,再开始处理节点(也就是在把节点的数值放进result数组中),这就造成了处理顺序和访问顺序是不一致的。

那么在使用迭代法写中序遍历,就需要借用指针的遍历来帮助访问节点,栈则用来处理节点上的元素。

我们设置cur为指针,指针帮助我们一层层访问左子树,直到叶子节点。而当到叶子节点则意味着我们要在做处理了,把根节点弹出,放入结果数组,再访问根节点的右子树,继续层层访问左子树,直到叶子节点。

class Solution {
public:vector<int> inorderTraversal(TreeNode* root) {vector<int> result;stack<TreeNode*> st;//用指针来访问元素TreeNode* cur = root;//当栈不为空, 当前根节点不为空时,循环以下操作while( cur == NULL||!st.empty()){// 如果指针不为空if(cur != NULL){st.push(cur);cur = cur->left;}//指针为空,说明要开始处理元素,则弹出栈顶else{//弹出根节点cur = st.top();st.pop();//放入结果数组result.push_back(cur->val);//访问根节点的右子树cur = cur->right;}  }return result;}
};

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/591950.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【mT5模型】mT5: A Massively Multilingual Pre-trained Text-to-Text Transformer

【mT5模型】mT5: A Massively Multilingual Pre-trained Text-to-Text Transformer 论文信息 阅读评价 Abstract Introduction Background on T5 and C4 mC4 and mT5 mC4 mT5 Comparison to related models Experiments Zero-shot generation Illegal predictions Pre…

基于SpringBoot+uniapp的同城活动报名系统开发找搭子软件

项目背景 随着移动互联网的飞速发展&#xff0c;人们的社交方式也在不断变化。在这个大背景下&#xff0c;同城活动报名系统应运而生&#xff0c;成为了连接人与人、活动与人之间的桥梁&#xff0c;深受广大年轻人的喜爱。在这个充满机遇与挑战的时代&#xff0c;同城活动报名…

【华为OD机试】山脉的个数、攀登者【C卷|100分】

【华为OD机试】-真题 !!点这里!! 【华为OD机试】真题考点分类 !!点这里 !! 题目描述 攀登者喜欢寻找各种地图,并且尝试攀登到最高的山峰。 地图表示为一维数组,数组的索引代表水平位置,数组的元素代表相对海拔高度。 其中数组元素0代表地面。 例如: [0,1,2,4,3,1,0,0,1,…

Dockerfile详解构建镜像

Dockerfile构建企业级镜像 在服务器上可以通过源码或rpm方式部署Nginx服务&#xff0c;但不利于大规模的部署。为提高效率&#xff0c;可以通过Dockerfile的方式将Nginx服务封装到镜像中&#xff0c;然后Docker基于镜像快速启动容器&#xff0c;实现服务的快速部署。 Dockerf…

clickhouse MPPDB数据库--新特性使用示例

clickhouse 新特性&#xff1a; 从clickhouse 22.3至最新的版本24.3.2.23&#xff0c;clickhouse在快速发展中&#xff0c;每个版本都增加了一些新的特性&#xff0c;在数据写入、查询方面都有性能加速。 本文根据clickhouse blog中的clickhouse release blog中&#xff0c;学…

黑马java-JavaWeb-Maven

1.Maven是专门用于管理和构建java项目的工具&#xff0c;它的主要功能有&#xff1a; 提供了一套标准化的项目结构提供了一套标准化的构建流程提供了一套依赖管理机制&#xff08;管理项目所依赖的第三方资源&#xff09; 2.Maven仓库 本地仓库&#xff1a;自己计算机上的一个目…

基于深度学习的植物病害检测系统(网页版+YOLOv8/v7/v6/v5代码+训练数据集)

摘要&#xff1a;本文深入研究了基于YOLOv8/v7/v6/v5的植物病害检测系统&#xff0c;核心采用YOLOv8并整合了YOLOv7、YOLOv6、YOLOv5算法&#xff0c;进行性能指标对比&#xff1b;详述了国内外研究现状、数据集处理、算法原理、模型构建与训练代码&#xff0c;及基于Streamlit…

Python打包成.exe可执行文件步骤

操作环境 python 3.7Anaconda Powershell Prompt 1、 安装 pyinstaller pip install pyinstaller 我使用的是anaconda&#xff1a; 2、cd到目标py文件的目录下 3、输入打包命令 pyinstaller -F -w dinosaur.py -F&#xff1a;打包后只生成单个exe格式文件&#xff1b;-D&am…

(免费分享)基于springboot,vue房屋租赁管理系统

功能说明&#xff1a; * 普通用户角色&#xff1a; 1. 寻找房源功能--提供了两种寻找房源的功能&#xff0c;一种是普通用户在平台上搜索、筛选主动寻找房源的功能&#xff0c;另一种是用户填写征集房源的条件&#xff0c;系统会持续将最新符合条件的房源推送给用户。 2. …

Vue tree自定义滚动条位置

贴一张效果图&#xff0c;我的效果不方便贴出来 实现支持&#xff1a; 1、懒加载 2、普通加载 下面贴关键思想&#xff1a; document有一个获取element元素的方法。 let element document.getElementById(tree); let arr document.querySelectorAll(".nodelModel&quo…

51单片机实验01-点亮LED小灯

目录 一&#xff0c;软件下载 二&#xff0c;单片机概述 1&#xff0c;单片机内部资源 1&#xff09;flash 2&#xff09;ram 3&#xff09;sfr 2&#xff0c;51单片机 3&#xff0c;单片机最小系统 三&#xff0c;点亮最右边的小灯 1&#xff0c;指出满足小灯点亮的有…

VsCode配置c++环境(详细)

这里写目录标题 1. 下载 MinGW添加环境变量 2. 配置VsCode3. 下载插件4. 测试5. 注意事项 1. 下载 MinGW MinGW 是 Minimalistic GNU for Windows 的缩写&#xff0c;相当于win版的gcc。 到MinGW官网下载。 也可以使用迅雷白嫖笔者下载好的 MinGW from 迅雷 提取码&#xff1a…