【刷题】：二叉树的中序遍历

原创已于 2022-07-20 15:26:39 修改 · 168 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #leetcode #数据结构 #算法

于 2022-07-08 10:38:05 首次发布

刷题库专栏收录该内容

5 篇文章

订阅专栏

本文解析了如何使用递归和迭代方法遍历二叉树的中序序列，通过实例和官方解释深入讲解了算法思路，适合理解二叉树结构和遍历方式的读者。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：给定一个二叉树的根节点 root ，返回 它的中序遍历

示例1：输入：root = [1,null,2,3] 输出：[1,3,2]

示例2：输入：root = [] 输出：[]

示例3：输入：root = [1] 输出：[1]

来源：力扣

/**
 * 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>nums;                    //设置存放结点的值的数组
        stack<TreeNode*>sta;                //设置存放遍历结点的指针的栈

        while(root!= nullptr || !sta.empty())   //指向二叉树的指针不为空或者栈不为空则循环
        {
            while(root != nullptr)             //指针不为空遍历左子树
            {
                sta.push(root);                //将指向左子树上结点的指针存入到栈内
                root = root->left;             //指针继续指向左节点
            }

            root = sta.top();                  //指针指向栈中顶部存放指针的指向的地方
            sta.pop();                         //弹出栈内的指针
            nums.push_back(root->val);         //将指针指向的结点上的值存储到数组中
            root = root->right;                //指针指向右节点
        }
        return nums;                           //返回数组
    }
};

解题思路：将指向左子树的指针存储到栈中，当遍历到空时，开始存储栈顶的指向的结点的值，当遇到存在右节点时，依旧循环遍历，直到栈内为空和节点为空，即遍历完成

class Solution {
public:
    void inorder(TreeNode* root, vector<int>& res) {
        if (!root) {
            return;
        }
        inorder(root->left, res);
        res.push_back(root->val);
        inorder(root->right, res);
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        inorder(root, res);
        return res;
    }
};

官方答案，递归的方法

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        TreeNode *predecessor = nullptr;

        while (root != nullptr) {
            if (root->left != nullptr) {
                // predecessor 节点就是当前 root 节点向左走一步，然后一直向右走至无法走为止
                predecessor = root->left;
                while (predecessor->right != nullptr && predecessor->right != root) {
                    predecessor = predecessor->right;
                }
                
                // 让 predecessor 的右指针指向 root，继续遍历左子树
                if (predecessor->right == nullptr) {
                    predecessor->right = root;
                    root = root->left;
                }
                // 说明左子树已经访问完了，我们需要断开链接
                else {
                    res.push_back(root->val);
                    predecessor->right = nullptr;
                    root = root->right;
                }
            }
            // 如果没有左孩子，则直接访问右孩子
            else {
                res.push_back(root->val);
                root = root->right;
            }
        }
        return res;
    }
};

详细解题思路说明，请点击官方解释

力扣https://leetcode.cn/problems/binary-tree-inorder-traversal/solution/er-cha-shu-de-zhong-xu-bian-li-by-leetcode-solutio/