&LeetCode94& 二叉树的中序遍历

题目

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

示例:
输入: [1,null,2,3]
1

2
/
3

输出: [1,3,2]

来源：力扣（LeetCode）

思路

二叉树的中序遍历顺序为左-根-右；
使用栈的解法：
首先，从根节点开始，先将根节点压入栈；
然后，再将其所有左子结点压入栈，然后取出栈顶节点，保存节点值；
其次，再将当前指针移到其右子节点上，若存在右子节点，则在下次循环时又可将其所有左子结点压入栈中。

C++代码

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) 
    {
        vector<int> res;
        stack<TreeNode*> s;
        TreeNode *p = root;

        while (p || !s.empty())
        {
            while (p)
            {
                s.push(p);
                p = p -> left;
            }
            p = s.top();
            s.pop();
            res.push_back(p -> val);
            p = p -> right;
        }
        return res;
    }
};