标签:node 面试题 tmpQueue 32 self None queue 二叉树 root
题目:
思路:
因为要求每层节点打印到一行,所以层次遍历时需要知道行的信息。个人思路通过两个队列的转换表示换行,优化思路记录当前层队列的长度。
代码:
Python
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# 队列交换
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
if root is None:
return []
levelQueue = []
tmpQueue = [root]
res = []
while tmpQueue:
# 给tmpQueue/tmpRes重新赋值[], 没有修改其元素不用deepcopy
levelQueue = tmpQueue
tmpQueue = []
tmpRes = []
for elem in levelQueue:
tmpRes.append(elem.val)
if elem.left is not None:
tmpQueue.append(elem.left)
if elem.right is not None:
tmpQueue.append(elem.right)
res.append(tmpRes)
return res
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# 队列长度
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
if root is None:
return []
queue = [root]
res = []
while queue:
tmp = []
for _ in range(len(queue)): # 这里len(queue)只在第一次计算, 之后不会变
node = queue.pop()
tmp.append(node.val)
if node.left:
queue.insert(0, node.left)
if node.right:
queue.insert(0, node.right)
res.append(tmp)
return res
相关问题
标签:node,面试题,tmpQueue,32,self,None,queue,二叉树,root 来源: https://www.cnblogs.com/cling-cling/p/13072870.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。