二叉树的层序遍历 宽度优先遍历

二叉树的层序遍历的理解一直模糊不清,虽然能写出来但是代码一一直很冗长!
对于普通的层序遍历不记录每一层有哪些数据的BFS很简单:

   s = [root]
   while s:
       cur_node = s.pop(0)
       print(cur_node.val)
       if cur_node.left:
           s.append(cur_node.left)
       if cur_node.right:
           s.append(cur_node.right)

但是,当我需要单独处理每一层的数据时,如何将每一层的数据分开就会比较麻烦,所以可以使用一个level数组来替换上面的s数组来存放每一层的数据,而不是一个一个的添加

    level = [root]
    while level:
        next_level = []
        # 直接以层位单位进行操作和添加!
        for i in level:
            #  当i为空时continue
            if not i:
                continue
            print(i.val)
            next_level.append(i.left)
            next_level.append(i.right)
        level = next_level
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值