力扣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 小时前
CSP-S 2021 T1廊桥分配
算法·图论
wangyue42 小时前
c# 线性回归和多项式拟合
算法
&梧桐树夏2 小时前
【算法系列-链表】删除链表的倒数第N个结点
数据结构·算法·链表
QuantumStack2 小时前
【C++ 真题】B2037 奇偶数判断
数据结构·c++·算法
今天好像不上班2 小时前
软件验证与确认实验二-单元测试
测试工具·算法
wclass-zhengge3 小时前
数据结构篇(绪论)
java·数据结构·算法
何事驚慌3 小时前
2024/10/5 数据结构打卡
java·数据结构·算法
结衣结衣.3 小时前
C++ 类和对象的初步介绍
java·开发语言·数据结构·c++·笔记·学习·算法
大二转专业5 小时前
408算法题leetcode--第24天
考研·算法·leetcode
凭栏落花侧5 小时前
决策树:简单易懂的预测模型
人工智能·算法·决策树·机器学习·信息可视化·数据挖掘·数据分析