【2024考研】题解16 | #求二叉树的层序遍历#

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <queue>
#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > levelOrder(TreeNode* root) {
        // write code here
        //二维数组存储
        vector<vector<int>> ans;
        //特例 空
        if(root == nullptr) return ans;
        //构造辅助队列q 根放进来
        queue<TreeNode*> q;
        q.push(root);
        //树节点指针定位当前节点
        TreeNode* cur;
        //只要队列里边不空 猛猛干
        while (!q.empty()) {
            vector<int> row;//临时行数组存储单层结点
            int n = q.size();//每层长度随着每一次循环而临时确定

            for(int i = 0; i < n; i++){
                cur = q.front();//当前结点指针定位到队头
                row.push_back(cur->val);//行数组暂时存储该节点值
                q.pop();//取值后的节点出队

                if(cur->left) q.push(cur->left);//注意这里不是递归
                if(cur->right) q.push(cur->right);
            }
            ans.push_back(row);//临时存储row传给最终ans输出
        }
        return ans;//最终输出
    }
};

时间复杂度：O(n)，其中n是二叉树的节点数。每个节点都会被遍历一次。

空间复杂度：O(m)，其中m是二叉树的最大宽度（即最多的节点数）。在队列中最多同时存储m个节点。

算法思想：使用队列进行层序遍历。

①首先将根节点入队，

②然后循环进行以下操作：从队列中取出当前层的所有节点，将它们的值存入临时行数组中，并将它们的左右子节点（如果有）入队。

③将临时行数组存入最终结果数组中。

④重复上述操作直到队列为空。

⑤最终输出最终结果数组。