【Leetcode HOT100】二叉树的中序遍历 c++

题目描述：

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

示例 1：

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

示例2:

输入：root = []
输出：[]

示例3:

输入：root = [1]
输出：[1]

提示：

树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

方法1 递归 c++代码：

/**
 * 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:
    void solve(TreeNode* root,vector<int> &v){
        if(root == nullptr)return;
        solve(root->left,v); //左
        v.push_back(root->val); //根
        solve(root->right,v); //右
    }
    //中序遍历：左 根 右
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> v;
        if(root==nullptr)return v;
        solve(root,v);
        return v;
    }
};

递归，中序遍历处理节点的顺序为：左子树，根，右子树。在递归函数中按照左子树，根节点，右子树的顺序处理即可。

方法2 迭代 c++代码：

/**
 * 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) {
        stack<TreeNode*> s;
        vector<int> v;
        if(root==nullptr)return v;
        while(root!=nullptr || !s.empty()){
            //入栈所有左子节点
            while(root!=nullptr){
                s.push(root);
                root = root->left;
            }
            //再把每个节点依次取出进行处理，记录节点值，再对当前节点的右子节点继续入栈所有的左子节点
            root = s.top();
            s.pop();
            v.push_back(root->val);
            root = root->right;
        }
        return v;
    }
};

用一个栈记录待处理节点。

若当前节点root不为空，或当前待处理节点栈不为空：

对当前节点root，进行while循环，入栈其所有最左子节点。
然后依次取出栈顶节点，记录该节点的值， 并对该节点的右子节点继续进行while循环，入栈所有其右子节点的所有最左子节点。

直到待处理栈为空，说明已经遍历完整个树了。