二叉树的中序遍历

二叉树的中序遍历

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

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

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;
    }
};

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

相关推荐
ZZZS05161 分钟前
stack栈练习
c++·笔记·学习·算法·动态规划
hans汉斯27 分钟前
【人工智能与机器人研究】基于力传感器坐标系预标定的重力补偿算法
人工智能·算法·机器人·信号处理·深度神经网络
vortex52 小时前
算法设计与分析:分治、动态规划与贪心算法的异同与选择
算法·贪心算法·动态规划
前端拿破轮2 小时前
🤡🤡🤡面试官:就你这还每天刷leetcode?连四数相加和四数之和都分不清!
算法·leetcode·面试
地平线开发者3 小时前
征程 6|工具链量化简介与代码实操
算法·自动驾驶
DoraBigHead3 小时前
🧠 小哆啦解题记——谁偷改了狗狗的台词?
算法
Kaltistss3 小时前
240.搜索二维矩阵Ⅱ
线性代数·算法·矩阵
轻语呢喃3 小时前
每日LeetCode:合并两个有序数组
javascript·算法
大熊猫侯佩4 小时前
Swift 数学计算:用 Accelerate 框架让性能“加速吃鸡”
算法·swift
杰克尼4 小时前
2. 两数相加
算法