LeetCode94 Binary Tree Inorder Traversal

Given a binary tree, return the inorder traversal of its nodes' values. （Medium）

For example:
Given binary tree [1,null,2,3],

   1
    \
     2
    /
   3

 

return [1,3,2].

Note: Recursive solution is trivial, could you do it iteratively?

 

分析：

太经典基础的算法问题了，但想写出一个无bug的非递归二叉树中序遍历也不是很容易。先看递归版本的代码：

class Solution {
private:
    vector<int> result;
    void helper(TreeNode* root) {
        if (root == nullptr) {
            return;
        }
        helper(root -> left);
        result.push_back(root -> val);
        helper(root -> right);
    }
public:
    vector<int> inorderTraversal(TreeNode* root) {
        helper(root);
        return result;
    }
};

再考虑非递归，其实就是对于每个节点，走到最左端，沿路径压栈。

到达最左端后以此返回，开始弹栈，对于每个弹出的元素，记录其value，并且走向其右节点重复上述过程（走到最左端...）。

直到栈内元素为空为止。

代码：

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> s;
        TreeNode* p = root;
        while (p || !s.empty()) {
            while (p != nullptr) {
                s.push(p);
                p = p -> left; 
            }
            if (!s.empty()) {
                p = s.top();
                result.push_back(p -> val);
                s.pop();
                p = p -> right;
            }
        }
        return result;
    }
};