leetcode剑指offer-BFS二叉树

面试题32.从上打印二叉树(1, 2 ,3)

从上打印二叉树.1
变体2
变体3
一刷:
1. 没有考虑root为None的情况
2. res_level在for中添加了导致输出出现错误
3. 没有正确的将res_level作为【】添加进res
第一次错误

#Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
from collections import deque
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]: 
    '''
    1 利用BFS,找到所有路径,注意BFS中的坑,比如向res中添加next_level添加的时机。
    2 不能使用列表作为now_level的容器,因为列表的pop(0)时间复杂度为O(n)
    3 deque不能直接用root创建需要变成list类型
    '''
    	if root is None:
    		return []
    	res = []
    	now_level = deque([root])
    	while now_level:
    		res_level = []
    		for _ in range(len(now_level)):
    			now_root = now_level.popleft()
    			if now_root.left is not None:
    				now_level.append(now_root.left)
    			if now_root.right is not None:
    				now_level.append(now_root.right)
    			res_level.append(now_root.val)
    		res.append(res_level)
    		#问题1:
    		#res = res + res_level
    	return res
    		

问题3:
有前两题的基础:用len(res)计算奇偶,改变res_level添加形式,就可以得出正确结果。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
from collections import deque
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if root is None:
            return []
        res = []
        now_level = deque([root])
        while now_level:
            res_level = deque([])
            for _ in range(len(now_level)):
                now_root = now_level.popleft()
                if (len(res) + 1) % 2 == 1:
                    res_level.append(now_root.val)
                else:
                    res_level.appendleft(now_root.val)
                if now_root.left is not None:
                    now_level.append(now_root.left)
                if now_root.right is not None:
                    now_level.append(now_root.right)
            res.append(list(res_level))
        return res

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值