【题目】
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
【代码】
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if not root:
return []
ans=[]
queue=[root]
while queue:
sz=len(queue)
temp=[]
for _ in range(sz):
root=queue.pop(0)
temp.append(root.val)
if root.left:
queue.append(root.left)
if root.right:
queue.append(root.right)
ans.append(temp)
return ans
【代码2】
class Solution:
def dfs(self,root,dep):
if not root:
return
if dep>len(self.ans):
self.ans.append([root.val])
else:
self.ans[dep-1].append(root.val)
self.dfs(root.left,dep+1)
self.dfs(root.right,dep+1)
def levelOrder(self, root: TreeNode) -> List[List[int]]:
self.ans=[]
self.dfs(root,1)
return self.ans