代码随想录Day14(3)|二叉树 统一迭代

前面用迭代法，难以实现统一的代码，不像递归法改一下节点顺序就可以转化成另一个。

统一的迭代遍历实现方法：

以中序遍历为例，由于无法同时解决访问节点（遍历节点）和处理节点（将元素放进结果集）不一致的情况。那么我们就将访问的节点放入栈中，把要处理的节点也放进栈中，但是要做好标记。

标记方式即，在把要处理的节点放入栈后，紧接着放空指针作为标记。这也叫标记法。

class Solution{
public:
   vector<int>inorderTraversal(TreeNode* root){
      vector<int>result;
      stack<TreeNode*>st;
      if(root != NULL)st.push(root);
      while(!st.empty()){
        TreeNode* node = st.top();
        if(node != NULL){
            st.pop();  //将该节点弹出，避免重复操作，下面再将右中左节点添加到栈中
            if(node->right)st.push(node->right);//添加右节点（空节点不入栈）

            st.push(node);                          //添加中节点
            st.push(NULL); //中节点访问过，但是还没有处理，加入空节点作为标记

            if(node->left)st.push(node->left);  //添加左节点（空节点不入栈）
        }else{                //只有遇到空节点的时候，才将下一个节点放进结果集
            st.pop();//将空节点弹出
            node = st.top();  //重新取出栈中元素
            st.pop();
            result.push_back(node->val);//加入到结果集
        }
      }
      return result;
   }
}

我们将访问的节点直接加入到栈中，但如果是处理的节点则后面放入一个空节点，这样只有空节点弹出的时候，才将下一个节点放进结果集。

迭代法前序遍历

class Solution{
public:
   vector<int>preordderTraversal(TreeNode* root){
      vector<int>result;
      stack<TreeNode*>st;
      if(root != NULL)st.push(root);
      while(!st.empty()){
        TreeNode* node = st.top();
        if(node != NULL){
            st.pop();
            if(node->right)st.push(node->right);
            if(node->lleft)st.push(node->left);
            st.push(node);
            st.push(NULL);
        }else{
            st.pop();
            node = st.top();
            st.pop();
            result.push_back(node->val);
        }
      }
      return result;
   }
}

迭代法后序遍历：

class Solution{
public:
  vector<int>postorderTraversal(TreeNode* root){
    vector<int>result;
    stack<TreeNode*>st;
    if(root != NULL)st.push(root);
    while(!st.empty()){
        TreeNode* node = st.top();
        if(node != NULL){
            st.pop();
            st.push(node);
            st.push(NULL);

            if(node->right)st.push(node->right);
            if(node->left)st.push(node->left);
        }else{
            st.pop();
            node = st.top();
            st.pop();
            result.push_back(node->val);
        }
    }
    return result;
  }
};