最優雅的二叉樹非遞歸遍歷模板

LeetCode 144. 二叉樹的前序遍歷

先壓棧的是右孩子，再是左孩子，因爲棧先進後出的特色，先取出來的就是左孩子，而後是右孩子，知足 根 - 左 - 右。

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

LeetCode 94. 二叉樹的中序遍歷

直接模擬中序遍歷的過程。

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

LeetCode 145. 二叉樹的後序遍歷

有點抽象，總之只需理解 根 - 右 - 左 逆過來就是 左 - 右 - 根。

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        if (!root)
            return res;
        stack<TreeNode*> s;
        s.push(root);
        TreeNode *node;
        while (!s.empty()) {
            node = s.top();
            s.pop();
            res.push_back(node->val);
            if (node->left)
                s.push(node->left);
            if (node->right)
                s.push(node->right);
        }
        reverse(res.begin(), res.end());
        return res;
    }
};