二叉树的定义

    1. struct TreeNode {
    2.     int val;
    3.     TreeNode *left;
    4.     TreeNode *right;
    5.     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
    6. };

    遍历

    1. class Solution {
    2. public:
    3.     //前序遍历
    4.     void pretraversal(TreeNode* cur, vector<int>& vec) {
    5.         if (cur == NULL) return;
    6.         vec.push_back(cur->val);    // 中
    7.         traversal(cur->left, vec);  // 左
    8.         traversal(cur->right, vec); // 右
    9.     }
    10.     //后续遍历
    11.     void posttraversal(TreeNode* cur, vector<int>& vec) {
    12.         if (cur == NULL) return;
    13.         traversal(cur->left, vec);  // 左
    14.         traversal(cur->right, vec); // 右
    15.         vec.push_back(cur->val);    // 中
    16.     }
    17.     //中序遍历
    18.     void intraversal(TreeNode* cur, vector<int>& vec) {
    19.         if (cur == NULL) return;
    20.         traversal(cur->left, vec);  // 左
    21.         vec.push_back(cur->val);    // 中
    22.         traversal(cur->right, vec); // 右
    23.     }
    25.     vector<int> preorderTraversal(TreeNode* root) {
    26.         vector<int> result;
    27.         traversal(root, result);
    28.         return result;
    29.     }
    30. };

    层序

    1. class Solution {
    2. public:
    3.     vector<vector<int>> levelOrder(TreeNode* root) {
    4.         queue<TreeNode*> que;
    5.         if (root != NULL) que.push(root);
    6.         vector<vector<int>> result;
    7.         while (!que.empty()) {
    8.             int size = que.size();
    9.             vector<int> vec;
    10.             // 这里一定要使用固定大小size，不要使用que.size()，因为que.size是不断变化的
    11.             for (int i = 0; i < size; i++) {
    12.                 TreeNode* node = que.front();
    13.                 que.pop();
    14.                 vec.push_back(node->val);
    15.                 if (node->left) que.push(node->left);
    16.                 if (node->right) que.push(node->right);
    17.             }
    18.             result.push_back(vec);
    19.         }
    20.         return result;
    21.     }
    22. };

    前中后迭代遍历

    1. //中
    2. class Solution {
    3. public:
    4.     vector<int> inorderTraversal(TreeNode* root) {
    5.         vector<int> result;
    6.         stack<TreeNode*> st;
    7.         if (root != NULL) st.push(root);
    8.         while (!st.empty()) {
    9.             TreeNode* node = st.top();
    10.             if (node != NULL) {
    11.                 st.pop(); // 将该节点弹出，避免重复操作，下面再将右中左节点添加到栈中
    12.                 if (node->right) st.push(node->right);  // 添加右节点（空节点不入栈）
    14.                 st.push(node);                          // 添加中节点
    15.                 st.push(NULL); // 中节点访问过，但是还没有处理，加入空节点做为标记。
    17.                 if (node->left) st.push(node->left);    // 添加左节点（空节点不入栈）
    18.             } else { // 只有遇到空节点的时候，才将下一个节点放进结果集
    19.                 st.pop();           // 将空节点弹出
    20.                 node = st.top();    // 重新取出栈中元素
    21.                 st.pop();
    22.                 result.push_back(node->val); // 加入到结果集
    23.             }
    24.         }
    25.         return result;
    26.     }
    27. };
    1. //前序只改上面几行
    2. if (node != NULL) {
    3.                 st.pop();
    4.                 if (node->right) st.push(node->right);  // 右
    5.                 if (node->left) st.push(node->left);    // 左
    6.                 st.push(node);                          // 中
    7.                 st.push(NULL);
    8. }
    9. //后序
    10. if (node != NULL) {
    11.                 st.pop();
    12.                 st.push(node);                          // 中
    13.                 st.push(NULL);
    14.                 if (node->right) st.push(node->right);  // 右
    15.                 if (node->left) st.push(node->left);    // 左
    17. }