0
点赞
收藏
分享

微信扫一扫

LeetCode--102.二叉树的层序遍历

乌龙茶3297 2022-02-16 阅读 95

二叉树的层序遍历

102二叉树的层序遍历

在这里插入图片描述
之前的二叉树的层序遍历利用队列进1个出1个,但是这里是把每一层放在一起。思路还是利用队列来解决。
分析如下图:
在这里插入图片描述
代码如下:

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root)
     {   
        vector<vector<int>> vv;
        //定义1个队列存结点 
        queue<TreeNode*> q;
       //树为空的情况
      if(root == nullptr)
      return vv;
     
     //第1层的结点个数为1个
      int leveSize = 1;
      //将root入队
      q.push(root);

      while(!q.empty())
      {         
          //用vector来存结点的值
          vector<int> v;
          //根据leveSize来控制每层出的个数 
          for(int i = 0; i < leveSize; ++i)
          {
              TreeNode* front = q.front();
              v.push_back(front->val);
              q.pop();
          //pop掉后,将front的左右入队
              if(front->left)
              {
                  q.push(front->left);
              }
                 if(front->right)
              {
                  q.push(front->right);
              }
          }
          //levelSize剩下的就是下一层的数据个数
          leveSize = q.size();
          vv.push_back(v);
      }
      return vv;
    }
};

就过了,时间复杂度为O(N).
在这里插入图片描述

107二叉树的层序遍历2

在这里插入图片描述
他这里要求我们从下往上输出,那我们直接将上面的代码考过来,在return 之前将vector逆置一下即可。
代码如下:

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root)
     {   
        vector<vector<int>> vv;
        //定义1个队列存结点 
        queue<TreeNode*> q;
       //树为空的情况
      if(root == nullptr)
      return vv;
     
     //第1层的结点个数为1个
      int leveSize = 1;
      //将root入队
      q.push(root);

      while(!q.empty())
      {         
          //用vector来存结点的值
          vector<int> v;
          //根据leveSize来控制每层出的个数 
          for(int i = 0; i < leveSize; ++i)
          {
              TreeNode* front = q.front();
              v.push_back(front->val);
              q.pop();
          //pop掉后,将front的左右入队
              if(front->left)
              {
                  q.push(front->left);
              }
                 if(front->right)
              {
                  q.push(front->right);
              }
          }
          //levelSize剩下的就是下一层的数据个数
          leveSize = q.size();
          vv.push_back(v);
      }
      return vv;
    }
};

在这里插入图片描述
过了,时间复杂度为O(N).

举报

相关推荐

0 条评论