二叉树的层次遍历

题目:

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

样例

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

  3
 / \
9  20
  /  \
 15   7

返回他的分层遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

思路:

      返回值是一个以向量为元素的向量,先在函数外定义两个向量std::vector<vector<int>> v2;和std::vector<int> v1;将一层的节点的数值放入v1,完成一个v1的放入就将v1为一个元素放入v2中,清空v1,便于下一层的节点的数进入v1中。建立一个队列暂存一层的结点,符合先进先出的原则。先将跟节点放入队列中,记录队列的节点的个数。最开始将根节点放入队列,长度为q.size(),并且循环这些次,保证了一层的节点的值进入同一个v1。将节点的数值放入v1中,如果当前节点的左节点不是空就放入队列,右节点相同,按q.soze()循环完放入v2,同时清空v1。当队列中再无节点时结束程序,反回v2。

代码:

class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: Level order a list of lists of integer
     */
public:
std::vector<vector<int>> v2;
std::vector<int> v1;
queue<TreeNode*> q;
TreeNode *top;
int l;
    vector<vector<int>> levelOrder(TreeNode *root) {
        // write your code here
       if(root==NULL)
       return v2;
       q.push(root);
       while(!q.empty())
       {
          l=q.size();
          while(l!=0)
          {
              top=q.front();
              v1.push_back(top->val);
              l--;
              if(top->left!=NULL)
              q.push(top->left);
              if(top->right!=NULL)
              q.push(top->right);
              q.pop();
          }
          v2.push_back(v1);
          v1.clear();
       }
       return v2;
    }
};

感想:

层次遍历用到队列是一个新的知识点,循环的方式也比较奇妙。这个题改过好几次思路,这个思路是最简洁的一个。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值