LeetCode 二叉树的遍历（迭代）和从前序与中序、中序与后序遍历序列构造二叉树

2020年8月19日 周三 天气晴 【不悲叹过去，不荒废现在，不惧怕未来】

---

---

2020.11.3 更新
剑指offer第7题就是这里的”从前序与中序遍历序列构造二叉树“，做的时候一点头绪都没有。。。
这里我竟然还用了递归和迭代两种做法。。。看来还是得经常复习 - -

---

这几个关于二叉树的问题都比较基础，但同时也非常重要，面试常考，要重点掌握。

关于二叉树的非递归遍历，网上有很多版本，也有人总结出了统一的模板，我仔细看了下那些模板，虽然统一了格式，但是相比于传统写法，每个结点都要进栈出栈两次，增加了运算量。因此我觉得最好还是记住这些常规的非递归写法比较好。

至于“从前序与中序、中序与后序遍历序列构造二叉树”这两个问题，一般也是用递归解决，而用迭代法的话，就要困难很多，但是万一面试的时候让你用迭代呢，毕竟递归也有它不好的地方，所以最好我们能掌握递归和迭代两种方法。

1. 二叉树的前序遍历（非递归）

首先把根结点入栈，然后在循环中执行以下操作：

• 此时栈顶元素为当前根结点，弹出并进行相关操作。
• 把当前根结点的右儿子和左儿子分别入栈（注意是右儿子先入栈左儿子后入栈，这样的话下次出栈的元素就是左儿子，才符合前序遍历的顺序要求：根结点->左儿子->右儿子）。

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

2. 二叉树的中序遍历（非递归）

循环里进行以下操作：

• 如果当前结点存在左结点，入栈。
• 否则处理栈顶结点，并令 指针（贯穿始终） 指向右结点。

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

3. 二叉树的后序遍历（非递归）

前面几行代码和和中序遍历一样，后面多了几行，就是判断根节点的右孩子是否被访问过，这里使用的方法是维护上一个节点 prev ，根据 root->right == prev? 来判断。

class Solution {
   
public:
    vector<int> postorderTraversal(TreeNode* root) {
   
        stack<TreeNode*> stk;
        TreeNode* prev = nullptr;
        vector<int> ans;
        while(root || !stk.empty()){
   
            // root存在的话，压入栈
            if(root){
   
                stk.push(root);
                root = root->left;
            }
            // root不存在，从栈顶取
            else{
   
                root = stk.top(); // 这里还不能直接弹出，因为可能右孩子还没访问，要先访问右孩子
                // 如果右孩子不为空，而且没有访问过，那就访问右孩子
                if(root->right && root->right!=prev){
   
                    root = root->right;
                }
                // 如果右孩子为空，或者已经被访问过了，那么就访问父节点
                else{
   
                    ans.push_back(root->val);
                    stk.pop();
                    prev