【代码随想录】LC 102. 二叉树的层序遍历

89 阅读2分钟

写在前面

本人CSDN博客主页:这里

一、题目

1、原题链接

力扣

2、题目描述

给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。

示例 1:

编辑

输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
示例 2:

输入:root = [1]
输出:[[1]]
示例 3:

输入:root = []
输出:[]

提示:

树中节点数目在范围 [0, 2000] 内
-1000 <= Node.val <= 1000

二、解题报告

思路来源:讲透二叉树的层序遍历 | 广度优先搜索 | LeetCode:102.二叉树的层序遍历_哔哩哔哩_bilibili

卡哥yyds

1、思路分析

1)可以利用队列来进行层序遍历。

2)如果根结点为空,则直接返回结果数组;如果根结点非空,在每次遍历一层的过程中,针对该层中的每个结点,首先将结点入队,然后将该结点存下来,结点出队,用记录结点的变量来访问该节点的数值,并存到一维数组中,如果该结点有孩子,则将其对应的左/右孩子结点入队,直到遍历完该层结点。然后再以同样方式遍历下一层结点,直到队列为空(代表已经遍历完所有元素)。用lq来记录当前层的结点个数,针对每层,将每层遍历的结点的值存入一维数组中,每遍历完一层,再将该数组存入二维结果数组中,用来结果的返回。

3)注意lq不能用q.size()来代替,因为队列大小一直在变化,队列中会进入其他层的结点,我们只是用lq来记录当前遍历层的结点个数。

4)模拟上述过程,返回结果数组,即为所求。

2、时间复杂度

时间复杂度为O(n)

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<vector<int>> levelOrder(TreeNode* root) {
      vector<vector<int>> res;
      queue<TreeNode*> q;
      if(root)
         q.push(root);
      vector<int> v;
      while(!q.empty()){
         int lq=q.size();
         while(lq--){
          TreeNode *temp=q.front();
          q.pop();
          v.push_back(temp->val);
          if(temp->left)
              q.push(temp->left);
          if(temp->right)
              q.push(temp->right);     
      }
      res.push_back(v);
      v.clear();
    }
    return res;
    }
};

三、知识风暴

clear();//删除vector容器中的所有元素