代码随想录算法训练营第十二天 | 递归遍历、迭代遍历、统一迭代（先不看）、层序遍历

1.递归遍历：

144.二叉树前序遍历（中左右）

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;

    }

};

145.二叉树后序遍历（就中间更换个顺序）（左右中）

void traversal(TreeNode* cur, vector<int>& vec) {
    if (cur == NULL) return;
    traversal(cur->left, vec);  // 左
    traversal(cur->right, vec); // 右
    vec.push_back(cur->val);    // 中
}

 94.二叉树的中序遍历（左中右）

void traversal(TreeNode* cur, vector<int>& vec) {
    if (cur == NULL) return;
    traversal(cur->left, vec);  // 左
    vec.push_back(cur->val);    // 中
    traversal(cur->right, vec); // 右
}

2.迭代遍历：

（1）前序遍历

 class Solution {

public:

    vector<int> preorderTraversal(TreeNode* root) {

        stack<TreeNode*> st;

        vector<int> result;

        if (root == NULL) return result;

        st.push(root);

        while (!st.empty()) {

            TreeNode* node = st.top();                       // 中

            st.pop();

            result.push_back(node->val);

            if (node->right) st.push(node->right);           // 右（空节点不入栈）

            if (node->left) st.push(node->left);             // 左（空节点不入栈）

        }

        return result;

    }

};

（2）中序遍历

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(); // 从栈里弹出的数据，就是要处理的数据（放进result数组里的数据）

                st.pop();

                result.push_back(cur->val);     // 中

                cur = cur->right;               // 右

            }

        }

        return result;

    }

};

（3）后序遍历

class Solution {

public:

    vector<int> postorderTraversal(TreeNode* root) {

        stack<TreeNode*> st;

        vector<int> result;

        if (root == NULL) return result;

        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;

    }

};

3.层序遍历：

# 递归法

class Solution {

public:

    void order(TreeNode* cur, vector<vector<int>>& result, int depth)

    {

        if (cur == nullptr) return;

        if (result.size() == depth) result.push_back(vector<int>());

        result[depth].push_back(cur->val);

        order(cur->left, result, depth + 1);

        order(cur->right, result, depth + 1);

    }

    vector<vector<int>> levelOrder(TreeNode* root) {

        vector<vector<int>> result;

        int depth = 0;

        order(root, result, depth);

        return result;

    }

};

## 非递归法

class Solution {

public:

    vector<vector<int>> levelOrder(TreeNode* root) {

        queue<TreeNode*> que;

        if (root != NULL) que.push(root);

        vector<vector<int>> result;

        while (!que.empty()) {

            int size = que.size();

            vector<int> vec;

            // 这里一定要使用固定大小size，不要使用que.size()，因为que.size是不断变化的

            for (int i = 0; i < size; i++) {

                TreeNode* node = que.front();

                que.pop();

                vec.push_back(node->val);

                if (node->left) que.push(node->left);

                if (node->right) que.push(node->right);

            }

            result.push_back(vec);

        }

        return result;

    }

};