迭代法遍历二叉树

前序遍历

题目来源：https://leetcode.cn/problems/binary-tree-preorder-traversal/description/

通过栈进行遍历，也就是访问元素后将其入栈，然后访问右节点，再访问左节点（出栈顺序相反），代码如下：

class Solution{
    public:
    vector<int> preorderTraversal(TreeNode* root){
        vector<int> result;
        stack<TreeNode*> st;
        if (root == nullptr) return result;
        st.push(root);
        while(!st.empty()){
            TreeNode * node = st.top();
            st.pop();
            result.push_back(node->val);
            if(node -> right) st.push(node -> right);
//注意这里要将节点放入栈，而不是节点的数值！
            if(node -> left) st.push(node -> left);

        }return result;
    }
};

举个例子来论证过程:给出一个二叉树[5,4,1,-1,-1,2,6,-1,-1],前序遍历顺序是中左右，也就是先遍历中介点，然后去左节点，右节点，顺序是54126，程序思路是做一个做一个栈来统计出入的值，做一个一维数组result存放遍历结果。首先把根节点入栈，然后判断根节点的值放入result，再求它右节点和左节点，此时把6和4按先后进栈，这个时候node为4，然后4出栈，完后处理思路和前面一样。

注意点：

1、栈类型是TreeNode ：

stack<typename> name;

//这里的typename可以是任何基本类型，例如 int、double、char、结构体等，也可以是STL标准容器，例如vector、set、queue等。

中序遍历：

和递归法不同，迭代法的中序遍历并不是前序遍历修改顺序，原因很简单，前序遍历的顺序是中左右，先遍历中节点然后是左右，正好我们入栈顺序也是中节点先进去，无非是左右节点交换进去的顺序以保证出栈的顺序相反。即要访问的元素和要处理的元素顺序是一致的，都是中间节点。但是中序遍历要求左中右，也就是先遍历左边的节点，然后中间节点，右节点。这个时候带来了问题，要访问的元素和要处理的元素顺序不一致。

那么再看看中序遍历，中序遍历是左中右，先访问的是二叉树顶部的节点，然后一层一层向下访问，直到到达树左面的最底部，再开始处理节点（也就是在把节点的数值放进result数组中），这就造成了处理顺序和访问顺序是不一致的。——代码随想录。

那么在使用迭代法写中序遍历，就需要借用指针的遍历来帮助访问节点，栈则用来处理节点上的元素。

代码如下，共性的解释见前序遍历迭代法。

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;//一维数组，用于存放遍历的节点数值
        stack<TreeNode*> st;
        TreeNode* cur = root;//建立指针，用于访问节点（指针访问的节点存到st中，和result没关系）
        while (cur != NULL || !st.empty()) {
            if (cur != NULL) { // 指针来访问节点，访问到最底层
                st.push(cur); // 将访问的节点放进栈
                cur = cur->left;                // 左
            } else {
                cur = st.top(); // 这个时候cur指向的是空，所以我们从栈顶的第一个值赋予cur，从栈里弹出的数据，就是要处理的数据（放进result数组里的数据）
                st.pop();
                result.push_back(cur->val);     // 中
                cur = cur->right;               // 右
            }
        }
        return result;
    }
};

通俗的语言来解释，那就是：给出一个二叉树[5,4,1,-1,-1,2,6,-1,-1]通过指针一路向左遍历同时入栈st，直到发现节点为空，也就是遍历到了二叉树最左边，这个时候指针cur指向了1的左孩子，此时节点为空，此时指针指向1，1出栈并存入result，指针指向1的右孩子，此时还是空，则再次出栈，指针指向出栈的数值所在节点.....

也就是一直往左遍历，遇到空就处理出栈中节点，然后往右，如果右节点不为空，那就继续往左走，为空则继续出栈。

后序遍历：

先序遍历是中左右，后续遍历是左右中，其中左右都是绑定在一起的，而不是中序的走中右。那么我们只需要调整一下先序遍历的代码顺序，就变成中右左的遍历顺序，然后在反转result数组，输出的结果顺序就是左右中了，如下图：

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

        }
        reverse(result.begin(), result.end()); // 结果反转
        return result;
    }
};