树的非递归遍历——前序、中序、后序

树的递归遍历非常简单，也是写树的遍历时最常用的写法。但是我发现自己对树的非递归遍历并不十分熟悉，所以把三种非递归遍历都写了一遍，以后看到这篇记录博客也可以帮助自己好好回想熟悉一下。

Leetcode对应习题：前序，中序，后序。

相对而言，这三种非递归遍历的难度——前序 < 中序 < 后序。

对于第三种(非递归后序遍历)方法，只要稍微调整下第18~19行三个节点push的顺序，就可以实现前中后序的遍历。

树的非递归前序：

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

树的非递归中序：

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

 

树的非递归后序遍历：

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<pair<TreeNode*, bool> > s;
        vector<int> re;
        s.push(make_pair(root, false));
        bool visited;
        while(!s.empty()){
            pair<TreeNode*, bool> p = s.top();
            s.pop();
            visited = p.second;
            if(!p.first){
                continue;
            }
            if(visited){
                re.push_back(p.first->val);
            }else{
                s.push(make_pair(p.first, true));
                s.push(make_pair(p.first->right, false));
                s.push(make_pair(p.first->left, false));
            }
        }
        return re;
    }
};