前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >69. 二叉树的层次遍历层次遍历+queue

69. 二叉树的层次遍历层次遍历+queue

作者头像
和蔼的zhxing
发布2018-09-04 11:35:03
1K0
发布2018-09-04 11:35:03
举报
文章被收录于专栏:和蔼的张星的图像处理专栏

给出一棵二叉树,返回其节点值的层次遍历(逐层从左往右访问)

样例

给一棵二叉树 {3,9,20,#,#,15,7}

代码语言:javascript
复制
  3
 / \
9  20
  /  \
 15   7

返回他的分层遍历结果:

代码语言:javascript
复制
[
  [3],
  [9,20],
  [15,7]
]

层次遍历+queue

参见数据结构与算法中写的,层次遍历是需要借助queue来做的,单纯的逐层遍历写起来是比较简单的,像这样要求不同的层还要放在不同的vector中,稍微难一点,我一开始也没想好到底怎么做,参考了别人的代码,实际上也不是很难,主要是记录一下每层的长度,那如何知道每一层的长度呢,用了一个很巧妙的方法。

  1. 首先建立一个队列que,把根节点放进去。伪代码:
代码语言:javascript
复制
while(que非空)
{
        建立vector<int>   x;
        int len=que.size();
       while(len--)
      {
            把front->val放进x,并删掉front;
            把front的左右子节点放入que;(先把front节点记录下来)            
      }
      把x放入vecto<vector<int>>  res ;
} 
返回  res;

这样操作的巧妙之处在于每次可以用len记录当前层的节点的个数,然后通过while循环把当前节点的下一层放进queue,这样while出来之后刚好是遍历完了这一层(而且已经删掉),queue里面剩下的就是下一层的节点了。

代码语言:javascript
复制
vector<vector<int>> levelOrder(TreeNode * root) {
        vector<vector<int>>  res;
        if(!root)
        return res;
        
        queue<TreeNode*>   que;
        que.push(root);
        int len=1;      //第一个节点的大小
        
        while(!que.empty())
        {
            len=que.size();
            vector<int> vtmp;
            
            
            while(len--)
            {
                TreeNode *tmp;
                tmp=que.front();
                vtmp.push_back(tmp->val);
                que.pop();
                if(tmp->left)   que.push(tmp->left);
                if(tmp->right)  que.push(tmp->right);
                
            }
            if(!vtmp.empty())
                res.push_back(vtmp);
        }
        return res;
        
        // write your code here
    }
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018.01.24 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 层次遍历+queue
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档