LintCode_067_二叉树的中序遍历

问题描述

给出一棵二叉树，返回其节点值的中序遍历。

样例
给出一棵二叉树 {1,#,2,3},

   1
    \
     2
    /
   3

返回 [1,3,2].

挑战

你能使用非递归实现么？

求解

class Solution
{
  public:
    vector<int> inorderTraversal(TreeNode *root)
    {
        vector<int> res;
        stack<pair<TreeNode *, bool>> s;
        s.emplace(root, false);
        while (!s.empty())
        {
            bool visited;
            tie(root, visited) = s.top();
            s.pop();
            if (root == nullptr)
            {
                continue;
            }
            if (visited)
            {
                res.emplace_back(root->val);
            }
            else
            {
                s.emplace(root->right, false);
                s.emplace(root, true);
                s.emplace(root->left, false);
            }
        }
        return res;
    }
};