代码随想录算法训练营第十四天 | 144.二叉树的前序遍历 94.二叉树的中序遍历 145.二叉树的后序遍历 二叉树的统一迭代法

144.二叉树的前序遍历

题目链接：力扣

递归法：

void traversal(TreeNode* root, vector<int>& vec) {
        if(root == nullptr) return;
        // 前序遍历 其他遍历只需要改变中左右的位置即可
        vec.push_back(root->val); // 中
        traversal(root->left, vec);// 左
        traversal(root->right, vec);// 右
    }
    vector<int> preorderTraversal(TreeNode* root) {
        // 递归法遍历二叉树
        vector<int> result;
        traversal(root, result);
        return result;
    }

迭代法：

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        // 迭代遍历（前序）
        // 基本思路：利用栈来实现迭代法，先将头结点加入栈，然后将栈弹出一个结点，将其值存入结果集，然后先判断其右结点是否为空，不为空加入栈，然后再判断左结点，一直重复，知道栈为空，其中注意，右结点先进的原因的栈先进后出，这时候左结点先出，我们遍历顺序是中左右，所以说要右结点先进
        vector<int> result;
        stack<TreeNode*> st;
​
        if(root != NULL) st.push(root);
        while(!st.empty()) {
            
            TreeNode* tmp = st.top();
            result.push_back(tmp->val);
            st.pop();
            if(tmp->right != NULL) st.push(tmp->right);
            if(tmp->left != NULL) st.push(tmp->left);
        }
        return result;
    }
};

94.二叉树的中序遍历

题目链接：力扣

​
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        // 迭代法（中序）
        // 基本思路：利用栈来解决，中序遍历和前序，后序不一样，不能使用其方法实现，具体实现为利用栈，先判断结点是否为空，不为空的话将其结点存入栈，然后将指针指向左结点，若为空，则取出栈顶元素，加入结果集，让指针指向其右结点，然后继续判断，知道栈和指针都为空则停止，返回结果
        vector<int> result;
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while(cur != NULL || !st.empty()) {
            if(cur != NULL) {
                st.push(cur);
                cur = cur->left;
            } else {
                cur = st.top();
                st.pop();
                result.push_back(cur->val);
                cur = cur->right;
            }
        }
        return result;
    }
};

145.二叉树的后序遍历

题目链接：力扣

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        // 迭代遍历（后序）
        // 基本思路：因为前序遍历是中左右，如果将遍历顺序转成中右左，然后将结果数组翻转，就成为了左右中
        // 这样就可以实现后序遍历二叉树
​
        vector<int> result;
        stack<TreeNode*> st;
​
        if(root == NULL) return result;
        
        st.push(root);
​
        while(!st.empty()) {
            TreeNode* tmp = st.top();
            result.push_back(tmp->val);
            st.pop();
            // 由于栈的特殊性，我们需要将左结点先加入这样遍历出来就是右结点在前
            if(tmp->left != NULL) st.push(tmp->left);
            if(tmp->right != NULL) st.push(tmp->right);
        }
        reverse(result.begin(), result.end());
​
        return result;
    }
};

二叉树的统一迭代法

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        // 统一迭代法的基本思路：由于要使用栈来同时完成二叉树的前、中、后序遍历，所以说在入栈的时候就要注意，先将头结点存入栈，然后在while循环里面，先将指针指向栈顶元素，判断元素是否为空，若不为空则需要先取出（防止重复操作），然后将其不为空的右，当前弹出元素和NULL、左来存入栈，这里存入NULL的原因是如在遇到NULL说明就需要将元素存入结果集了，想要前序就需要依次放入右，右、左、中，因为栈是先入后出，这样一直循环就可以做到用一种方法实现二叉树的前、中、后序遍历
        vector<int> result;
        stack<TreeNode*> st;
​
        if(root == NULL) return result;
​
        st.push(root);
​
        while(!st.empty()) {
            TreeNode* tmp = st.top();
            if(tmp != NULL) {
                st.pop();
                //只需要改变下列代码的顺序即可
                if(tmp->right) st.push(tmp->right);
                st.push(tmp);
                st.push(NULL);
                if(tmp->left) st.push(tmp->left);
                
            } else {
                st.pop();
                tmp = st.top();
                result.push_back(tmp->val);
                st.pop();
            }
        }
        return result;
    }
};