LeetCode 94-Binary Tree Inorder Traversal（二叉树，中序遍历的迭代）

LeetCode 94-Binary Tree Inorder Traversal

题干：

二叉树的中序遍历，用迭代写。

解：

方法1：

从根节点开始一直往下访问节点的左孩子到最底层（同时压入栈），如果往下访问不了了，那么取出栈顶元素把值放入ans数组，再访问它的右孩子，继续上述操作。

vector<int> inorderTraversal(TreeNode* root) {
        if(!root) return {};
        vector<int> ans;
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while(cur || !st.empty()){
            if(cur){ // 指针来访问节点，访问到最底层
                st.push(cur); 
                cur = cur->left; //左
            }
            else{
                cur = st.top();
                st.pop();
                ans.push_back(cur->val); //中
                cur = cur->right; //右
            }
        }
        return ans;
    }

方法2：

统一写法。

将访问的节点按递归顺序放入栈中，把要处理的节点也放入栈中但是做好标记。

如何标记？就是要处理的节点放入栈后，紧接着放入一个空指针作为标记。

vector<int> inorderTraversal(TreeNode* root) {
        if(!root) return {};
        stack<TreeNode*> st;
        vector<int> ans;
        st.push(root);
        while(!st.empty()){
            TreeNode* tmp = st.top();
            st.pop(); // 将该节点弹出，避免重复操作
            if(tmp){
                if(tmp->right) st.push(tmp->right); // 右
                st.push(tmp); // 中
                st.push(nullptr);
                if(tmp->left) st.push(tmp->left); // 左
            }
            else{
                tmp = st.top();
                st.pop();
                ans.push_back(tmp->val);
            }
        }
        return ans;
    }

参考