leetcode hot100刷题日记——33.二叉树的层序遍历

解题+总结二维vector的初始化方法

题目描述


解答:用队列

思路都差不多,我觉得对于我自己来说,此题最大的难点在于二维vector的初始化

和之前有行数的初始化不一样的是:此题不确定行数,也不确定列数

那我们就以此题为例,总结一下二维vector的初始化吧。

情况1:不确定行数和列数

cpp 复制代码
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        //用队列啦(先进先出)
        //每一层节点都进队列完,再开始一个一个出队列
        //出来的队头元素继续看有没有左右孩子节点,有的话进队列,没有的话那就不管了
        vector<vector<int>>res;
        if(!root){
            return res;
        }
        queue<TreeNode*>q;
        q.push(root);
        while(!q.empty()){
            int curLevelsize=q.size();//记录当前层有的节点个数
            //vector<int>()创建一个空的vector<int>对象。push_back将这个空的vector<int>添加到res的末尾。每次调用该语句时,都会为新的一层创建一个空的vector<int>,用于存放该层的节点值
            res.push_back(vector<int>());
            for(int i=0;i<curLevelsize;i++){
                TreeNode *node=q.front();
                q.pop();
                //res.back()获取res中最后一个vector<int>(即当前层的存储容器)。push_back(node->val)将当前节点的值添加到该vector<int>的末尾。
                res.back().push_back(node->val);
                if(node->left){
                    q.push(node->left);
                }
                if(node->right){
                    q.push(node->right);
                }
            }
            
        }
        return res;
    }
};

定义:vector<vector>res;

初始化新的行并且先插进vector中:res.push_back(vector());

在新的行中一个一个添加数:res.back().push_back(node->val);

cpp 复制代码
vector<vector<int>> levelOrder(TreeNode* root) {
    vector<vector<int>> res;
    if (!root) return res;
    queue<TreeNode*> q;
    q.push(root);
    while (!q.empty()) {
        int curLevelSize = q.size();
        res.emplace_back(); // 直接在res末尾构造一个空的vector<int>
        for (int i = 0; i < curLevelSize; i++) {
            TreeNode* node = q.front();
            q.pop();
            res.back().push_back(node->val); // 添加到最后一层的vector中
            if (node->left) q.push(node->left);
            if (node->right) q.push(node->right);
        }
    }
    return res;
}

定义:vector<vector> res;

初始化新的行并且先插进vector中:res.emplace_back();

再在新的行中一个一个添加数:res.back().push_back(node->val);

(此法效率比上一个高些)

cpp 复制代码
vector<vector<int>> levelOrder(TreeNode* root) {
    vector<vector<int>> res;
    if (!root) return res;
    queue<TreeNode*> q;
    q.push(root);
    while (!q.empty()) {
        int curLevelSize = q.size();
        vector<int> level(curLevelSize); // 预留当前层的空间大小
        for (int i = 0; i < curLevelSize; i++) {
            TreeNode* node = q.front();
            q.pop();
            level[i] = node->val; // 直接赋值
            if (node->left) q.push(node->left);
            if (node->right) q.push(node->right);
        }
        res.push_back(level); // 将当前层的vector整体添加到res中
    }
    return res;
}

定义:vector<vector> res;

先单独初始化行:vector level(curLevelSize);

再把整行插入vector:res.push_back(level);

(个人感觉此方法最好记)

情况2:已知行数和列数

cpp 复制代码
int rows = 3;
int cols = 4;
vector<vector<int>> res(rows, vector<int>(cols));

res[i][j] = value;  // 直接为第 i 行第 j 列赋值

情况3:已知行数但不知道列数

cpp 复制代码
int rows = 5;
vector<vector<int>> res(rows);

vector<int> temp = {1, 2, 3};
res[i] = temp;  // 将第 i 行设置为新行
res[i].push_back(4);  // 在第 i 行中添加元素4

情况4:已知列数但不知道行数

cpp 复制代码
int cols = 4;
vector<vector<int>> res;

vector<int> row(cols);
res.push_back(row);  // 添加一行,每行有cols个默认初始化的元素
相关推荐
ZTLJQ8 分钟前
专业Python爬虫实战教程:逆向加密接口与验证码突破完整案例
开发语言·数据结构·爬虫·python·算法
我有一计33310 分钟前
【算法笔记】6.LeetCode-Hot100-链表专项
人工智能·算法·程序员
努力的小帅1 小时前
C++_红黑树树
开发语言·数据结构·c++·学习·算法·红黑树
CoovallyAIHub1 小时前
YotoR模型:Transformer与YOLO新结合,打造“又快又准”的目标检测模型
深度学习·算法·计算机视觉
孟大本事要学习1 小时前
算法第29天|动态规划dp2:不同路径、不同路径Ⅱ、整数拆分、不同的二叉搜索树
算法·动态规划
ezl1fe1 小时前
RAG 每日一技(九):你的RAG是“人工智障”还是“真智能”?聊聊RAG的评估
后端·算法
এ᭄画画的北北2 小时前
力扣-51.N皇后
算法·leetcode
qqxhb2 小时前
零基础数据结构与算法——第六章:算法设计范式与高级主题-设计技巧(上)
java·数据结构·算法·分解·空间换时间·时空平衡
1白天的黑夜12 小时前
前缀和-974.和可被k整除的子数组-力扣(LeetCode)
c++·leetcode·前缀和
2 小时前
LeetCode Hot 100 搜索二维矩阵
算法·leetcode·矩阵