Leetcode 144-Binary Tree Preorder Traversal（二叉树，前序遍历的迭代写法）

Leetcode 144-Binary Tree Preorder Traversal

题干：

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

解：

方法1：

用栈模拟递归过程。

先把根节点放入栈中。

取出栈顶元素，把它的值放到ans数组内，然后把它的右孩子加入栈，再加入左孩子；循环这个操作直到栈为空。

先加入右孩子再加入左孩子的目的是为了保证出栈是中左右的顺序。

注意空节点不入栈。

vector<int> preorderTraversal(TreeNode* root) {
        if(!root) return {};
        vector<int> ans;
        stack<TreeNode*> st;
        st.push(root);
        while(!st.empty()){
            TreeNode* t = st.top(); //中
            st.pop(); 
            ans.push_back(t->val);
            if(t->right) st.push(t->right);  // 右（空节点不入栈）
            if(t->left) st.push(t->left); // 左（空节点不入栈）
        }
        return ans;
    }

方法2：统一写法

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

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

vector<int> preorderTraversal(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); //右
                if(tmp->left) st.push(tmp->left); //左
                st.push(tmp); //中
                st.push(nullptr);
            }
            else{
                tmp = st.top();
                st.pop();
                ans.push_back(tmp->val);
            }
        }
        return ans;
    }

参考