二叉树的中序遍历

二叉树的中序遍历

法一:递归,最简单的方法

直接上代码:注意 递归终止条件为碰到空结点。

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 {
    vector<int>ans;
    void zxbl(TreeNode* node)
    {
        if(!node) return;//注意递归都要有终止条件,这里递归终止的条件为碰到空节点。
        zxbl(node->left);
        ans.push_back(node->val);//加入当前结点
        zxbl(node->right);
    }
public:
    vector<int> inorderTraversal(TreeNode* root) {
        zxbl(root);
        return ans;
    }
};

法二:迭代(要掌握!)

递归函数我们也可以用迭代的方式实现,两种方式是等价的,区别在于递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来,其他都相同,具体实现可以看下面的代码。

关键思路就是这个栈(先入后出)! 首先对于每个结点 就先放入栈中,然后看它有没有左孩子,有左孩子就将左孩子继续放入栈中,没有左孩子就把当前结点输出,再看它有没有右孩子,有的话就放入栈中,继续上述操作,每次都处理当前结点。

代码其实之前有写过类似的。

注意:栈和队列是push,vector是push_back

代码:

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> ans;
        stack<TreeNode *> st;//栈,存每一个结点,是指针类型的
        while(root||!st.empty())//栈不空
        {
            while(root)
            {
                st.push(root);
                root=root->left;
            }
            auto t=st.top();
            st.pop();
            ans.push_back(t->val);
            root=t->right;

        }
        return ans;
    }
};

这个迭代的代码需要好好理解,为什么这么写,好好思考琢磨。

相关推荐
不如自挂东南吱25 分钟前
空间相关性 和 怎么捕捉空间相关性
人工智能·深度学习·算法·机器学习·时序数据库
荒诞硬汉39 分钟前
对象数组.
java·数据结构
洛生&1 小时前
Elevator Rides
算法
2501_933513041 小时前
关于一种计数的讨论、ARC212C Solution
算法
Wu_Dylan1 小时前
智能体系列(二):规划(Planning):从 CoT、ToT 到动态采样与搜索
人工智能·算法
散峰而望1 小时前
【算法竞赛】栈和 stack
开发语言·数据结构·c++·算法·leetcode·github·推荐算法
知乎的哥廷根数学学派1 小时前
基于多尺度注意力机制融合连续小波变换与原型网络的滚动轴承小样本故障诊断方法(Pytorch)
网络·人工智能·pytorch·python·深度学习·算法·机器学习
蚊子码农1 小时前
算法题解记录-208实现Trie前缀树
运维·服务器·算法
2301_800256111 小时前
【人工智能引论期末复习】第3章 搜索求解2 - 对抗搜索
人工智能·算法·深度优先
程序猿阿伟1 小时前
《量子算法开发实战手册:Python全栈能力的落地指南》
python·算法·量子计算