代码随想录算法训练营day14||前序，中序，后序遍历

前序，中序，后序遍历对应图论中的DFS，主要有两种方法：一种是递归法，一种是迭代法。由于面试中一般都会考察迭代法做简单的前中后序遍历，因此今天主要练习的代码也是用迭代法。不过作为初学者，要牢记递归法3部曲：

1.确定递归参数和返回值

2.确定终止条件

3.确定单层递归的逻辑

迭代法又有一般迭代和统一的前中后序迭代的区别。在一般迭代中，前序遍历先获取中间节点值放入数组，再把右节点和左节点压入栈，然后在下面的循环中弹出，这样出来的就是中左右的顺序；后序遍历是在前序遍历的基础上，把左节点和右节点压入栈（这样出来的是中右左的顺序，再翻转一下数组，得到左右中的顺序），中序遍历复杂的原因是访问节点和要处理的节点不一致：访问节点从中间节点开始，处理要从左下的第一个节点开始。由于左下的第一个节点没有左孩子，因此先把指针通过不等于空的循环条件遍历到左下角第一个节点，这时候左子树已经被遍历过，需要遍历右子树，再去遍历右子树，如果没有右子树就弹出这个节点保存到数组里（也就是返回上一个节点）

统一的迭代法主要思想是把要访问的节点放入栈中，把要处理的节点（也就是中间节点）和一个标记节点null一起放入栈中，判断节点是否为空，当放入最后一个节点时，会再放入空节点，然后一个个弹出。由于统一的迭代法更方便但是需要练习，因此我在前序和中序遍历都是用统一迭代练手

lc.144:前序遍历

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != nullptr) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node != nullptr) {
                st.pop();
                if (node->right) st.push(node->right);
                if (node->left) st.push(node->left);
                st.push(node);
                st.push(nullptr);
            }else {
                st.pop();
                node = st.top();
                st.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};

lc.94:中序遍历

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != nullptr) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node != nullptr) {
                st.pop(); // 将获取节点弹出，之后按右中左的形式入栈
                if (node->right) st.push(node->right);
                st.push(node);
                st.push(nullptr);
                if (node->left) st.push(node->left);
            }else {
                st.pop();
                node = st.top();
                st.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};

lc.145:后序遍历

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != nullptr) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            st.pop();
            result.push_back(node->val);
            if (node->left) st.push(node->left);
            if (node->right) st.push(node->right);
        }
        reverse(result.begin(), result.end());
        return result;
    }
};