[M二叉树] lc94. 二叉树的中序遍历(栈+dfs)

1. 题目来源

链接：lc94. 二叉树的中序遍历

2. 题目说明

3. 题目解析

中序：左+根+右

几个遍历的 stack 写法，建议直接背过，可拓展性不强，在二叉树的情况下才会用到

变形多、可由模板扩展的算法，理解思路，背好模板
变形少、强固定且常用的算法，直接背过就行了

对代码做了注释，中序迭代写法我很容易出错…

说到中序遍历，那么提一提 前驱节点 ，后继节点 的定义吧：

• 前驱节点：对一棵二叉树进行中序遍历，在遍历后的顺序下，当前节点的前一个节点为该节点的前驱节点
• 后继节点：对一棵二叉树进行中序遍历，在遍历后的顺序，当前节点的后一个节点为该节点的后继节点

大家都知道二叉搜索树中序遍历为有序序列，这也是中序遍历很重要的应用。

友情链接：

[M二叉树] lc102. 二叉树的层序遍历(队列+bfs)
[M二叉树] lc145. 二叉树的后序遍历(栈+dfs)
[M二叉树] lc94. 二叉树的中序遍历(栈+dfs)
[M二叉树] lc144. 二叉树的前序遍历(栈+dfs)

迭代代码：

/**
 * 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> res;
        stack<TreeNode*> stk;

        while (root || stk.size()) {    // 当前节点或栈非空时
            while (root) {              // 当前节点非空,令当前节点入栈,当前节点向左走
                stk.push(root);
                root = root->left;
            }                           // while循环结束则root一定为空
            // 在此的stk一定不会空,因为如果至此root为空,则第一个while循环还能进来的话,一定栈非空
            // 如果栈为空,则root一定不为空,则在第二个while循环中就将root加了进来
            // 所以栈一定非空，可以优化掉这个判断
            if (stk.size()) {           // 找后继节点即为栈顶元素,需提前判断栈非空
                auto t = stk.top();     // 栈顶即为后继节点
                stk.pop();
                res.push_back(t->val);  
                root = t->right;        // 走到当前节点的右节点
            }
        }

        return res;
    }
};

递归代码：

/**
 * 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> vt;
    vector<int> inorderTraversal(TreeNode* root) {
        if (root) {
            inorderTraversal(root->left);
            vt.push_back(root->val);
            inorderTraversal(root->right);
        }
        return vt;
    }
};