力扣94. 二叉树的中序遍历

题目链接

cpp 复制代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
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;
    }
};

好的,让我们用费曼学习法来理解这段中序遍历二叉树的代码,简化成易于理解的语言。

中序遍历的顺序

首先,中序遍历的顺序是:左-中-右。意味着对于任何节点,先访问其左子树,然后访问该节点本身,最后访问其右子树。

思路概述

我们使用一个栈来帮助我们模拟这个过程,因为栈具有"后进先出"的特性,可以帮助我们保持访问顺序。同时,我们使用一个指针cur来遍历树。

分步解析

  1. 初始化 :创建一个空栈st和一个指向根节点的指针cur

  2. 遍历条件 :当cur不是空的,或者栈不是空的时,我们继续遍历。这意味着只要我们有节点要处理,或者栈中还有待处理的节点,就继续。

  3. 向左深入 :如果cur不为空,我们将cur推入栈中,并将cur更新为其左子节点。这个过程会一直持续到没有左子节点为止,即我们到达了某个分支的最左边。

  4. 访问节点 :当我们无法再向左深入时(cur为空),我们从栈中取出一个节点(这实际上是我们最后访问的那个节点的父节点),并将该节点的值加入到结果列表中。这是"中"部分的操作,因为我们此时访问了这个节点。

  5. 向右转 :之后,我们将cur更新为刚才处理的节点的右子节点。如果有右子节点,下一个循环会继续深入到这个右子节点的最左边。如果没有右子节点,下一次循环会从栈中取出下一个节点来处理。

重点理解

  • 栈的作用:栈帮助我们管理回溯。每当我们深入到一个分支的最左边,栈就保存了一条路径上的所有节点。这样,当我们需要回溯时,可以直接从栈中取出上一个节点。

  • 左-中-右遍历:通过在遍历到最左边后处理节点(加入结果列表),然后转向右子节点,我们保证了遍历的顺序是左-中-右。

  • 循环条件while循环的条件cur != NULL || !st.empty()确保了只要我们还有节点要处理,就不会停止。这包括两种情况:一是当前节点非空,需要继续深入;二是当前节点为空,但栈中还有待处理的节点。

通过这样的方法,我们不需要递归,就能以迭代的方式实现中序遍历,这对于理解栈如何在算法中用于管理状态很有帮助。希望这样解释后,你能更好地理解和记住这段代码!

相关推荐
徐浪老师1 小时前
C语言实现冒泡排序:从基础到优化全解析
c语言·算法·排序算法
hr_net1 小时前
图论入门编程
算法·图论
李小白661 小时前
各种排序算法
数据结构·算法·排序算法
浪前1 小时前
排序算法之冒泡排序篇
数据结构·算法·排序算法
小黄编程快乐屋1 小时前
各个排序算法基础速通万字介绍
java·算法·排序算法
PeterClerk1 小时前
图论基础知识
算法·深度优先·图论
是糖不是唐1 小时前
代码随想录算法训练营第五十八天|Day58 图论
c语言·算法·图论
Eric.Lee20213 小时前
数据集-目标检测系列- 装甲车 检测数据集 armored_vehicles >> DataBall
python·算法·yolo·目标检测·装甲车检测
慢慢来_5 小时前
【力扣热题100】[Java版] 刷题笔记-448. 找到所有数组中消失的数字
笔记·算法·leetcode
橘子遇见BUG7 小时前
算法日记 33 day 动态规划(打家劫舍,股票买卖)
算法·动态规划