二叉树的遍历前中后（递归、非递归统一写法）

题目链接

1. 前序
2. 中序
3. 后序

前序

递归

/**
 * 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> df(TreeNode* root, vector<int> &res) {
        if (root != NULL) {
            res.push_back(root->val);
        }
        if (root->left != NULL) {
            df(root->left, res);
        }
        if (root->right != NULL) {
            df(root->right, res);
        }
       
        return res;
    }
    vector<int> preorderTraversal(TreeNode* root) {
        // initial ??
        vector<int> res;
        if (root == NULL) {
            return res;
        }
        return df(root, res);
    }
};
// 总结代码不够精炼
// 并且在写的时候没有考虑到一些空指针的情况
// 别人代码都是从空开始的，我是从非空开始的，所以导致代码看着很别扭

// 别人的代码
class Solution {
public:
    void traversal(TreeNode* cur, vector<int>& vec) {
        if (cur == NULL) return;
        vec.push_back(cur->val);    // 中
        traversal(cur->left, vec);  // 左
        traversal(cur->right, vec); // 右
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root, result);
        return result;
    }
};

迭代统一代码

/**
 * 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:

    // 迭代法(统一风格)7:20
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> st;
        if (!root) return res;
        st.push(root);
        // 中左右的顺序遍历，右左中入栈
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node) {
                st.pop();// 出
                if (node->right) st.push(node->right);
                if (node->left) st.push(node->left);
                st.push(node);// 入
                st.push(NULL);
            }
            else {
                st.pop();
                node = st.top();
                st.pop();
                res.push_back(node->val);
            }
        }
        return res;
    }
};

后序和中序

• 同理只是将入栈顺序改变一下