leetcode101 二叉树的层次遍历
这道题相当于标准DFS的一点变化。
题目如下
给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
双队列+DFS题解
开这道题目之前,我一看,这不是标准的DFS吗,为什么会归为Medium这个难度,后来仔细一想,这个题目需要把我们通过队列的值记录下来,一开始没有想到怎么样才可以记录下这些通过的值,后来自己思考了一会,想到了之前二叉树的最大深度和二叉树的最小深度这两道题的栈解法,就是使用两个栈来实现记忆数字,这样就得到了我们的题解。使用两个队列来实现节点和层数的记忆化。这样可以把时间复杂度从变成。这次是拿python写的,因为java的队列结构操作还不是很熟。代码如下
执行用时 : 24 ms, 在Binary Tree Level Order Traversal的Python提交中击败了100.00% 的用户
内存消耗 : 12.1 MB, 在Binary Tree Level Order Traversal的Python提交中击败了33.50% 的用户
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
if root == None:
return []
#使用下面两个队列来辅助实现
nodeQueue = [root]
numberQueue = [1]
ans = [[root.val]]
while nodeQueue:
node = nodeQueue.pop(0)
number = numberQueue.pop(0)
if node.left != None:
nodeQueue.append(node.left)
numberQueue.append(number+1)
try:
# print(node.left.val)
ans[number]+=[node.left.val]
except:
# print(node.left.val)
ans.append([node.left.val])
if node.right != None:
nodeQueue.append(node.right)
numberQueue.append(number+1)
try:
# print(node.right.val)
ans[number]+=[node.right.val]
except:
# print(node.right.val)
ans.append([node.right.val])
return ans