算法题打卡day14 | 二叉树的递归遍历和迭代遍历

二叉树的结构要会写：

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:
    void inorder(TreeNode* root, vector<int>& res){
        if(root == nullptr){
            return;
        }
        inorder(root->left, res);
        res.push_back(root->val);
        inorder(root->right, res);
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        inorder(root, res);

        return res;
    }
};

迭代遍历的方法思路上是利用栈来实现函数调用的效果，但是在具体实现时，前序和后序的思路类似，前序遍历通过“先处理栈顶节点，将右节点入栈，将左节点入栈”的顺序进行处理，后序的结果通过reverse()前序遍历的思路来得到，注意左右节点的入栈顺序是反的，因为前序的入栈序列是“右左中”，这样处理的顺序就是“中左右”；而后序的期望顺序是“左右中”，为此需要逆转的“前序”顺序是“中右左”，进而可以推导出入栈的顺序是“左右中”，和前序遍历的左右子节点入栈顺序相反。这里用后序遍历为例：

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> st;
        if(root == nullptr) { return res; }
        st.push(root);
        while(!st.empty()){
            TreeNode* node = st.top();
            st.pop();
            res.push_back(node->val);
            if(node->left) { st.push(node->left); } // 相对前序遍历，入栈的顺序要改一下。
            if(node->right) { st.push(node->right); }
        }
        reverse(res.begin(), res.end());
        return res;
    }
};

而中序遍历的思路和前序不同，因为它需要先处理最左边的节点，为此需要加入一个指针来对节点是否为空进行判断。一路向左走到底，接着当节点满足是最左边的节点时再进行处理。代码如下：

class Solution {
public:
    // 迭代，利用栈来实现函数调用相同的效果
    vector<int> inorderTraversal(TreeNode* root){
        vector<int> res;
        stack<TreeNode*> st;
        TreeNode* cur = root;

        while(cur != nullptr || !st.empty()){
            if(cur != nullptr){
                st.push(cur);
                cur = cur->left;
            }else{
                cur = st.top();
                st.pop();
                res.push_back(cur->val);
                cur = cur->right;
            }
        }
        return res;
    }
};