二叉树的前中后序迭代形式遍历

前序遍历

遍历顺序：根左右。先弹出栈顶访问根节点，接着利用栈先将右边子树存储，再存储左子树。

/**
 * 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> preorderTraversal(TreeNode* root) {
        if(root == nullptr) return {};
        stack<TreeNode*>stk;
        stk.push(root);
        vector<int>res;
        while(!stk.empty()){
            TreeNode* p = stk.top();
            stk.pop();
            res.push_back(p->val);
            if(p->right) stk.push(p->right);
            if(p->left) stk.push(p->left);
        }
        return res;
    }
};

后续遍历

遍历顺序：左右根。由于和前序遍历的根左右的逆序稍有不同，故可以稍微修改下前序比遍历的代码，遍历顺序改为根右左，最后将结果逆序即变为了左右根的遍历顺序。

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

中序遍历

遍历顺
序：左根右。有了根节点，先得处理它的左子树。

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