题目:给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
解法一:迭代法
- 用两个数组,分别表示当前遍历层temp、存储当前层的下一层temp2;
- 再用两个数组,分别存储当前层的值res1、总的各层值res2。
代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if not root:return []
res = []
temp = []
temp.append(root)
while temp:
res2 = []
temp2 = []
for cur in temp:
res2.append(cur.val)
if cur.left:
temp2.append(cur.left)
if cur.right:
temp2.append(cur.right)
res.append(res2)
temp = temp2
return res
执行结果:通过显示详情
执行用时 :36 ms, 在所有 python3 提交中击败了96.10%的用户
内存消耗 :12.8 MB, 在所有 python3 提交中击败了99.73%的用户
原本想用队列,可搞不清楚python3中的deque和queue的用法和区别。
在 Python 中如果使用 Queue 结构,但因为它是为多线程之间安全交换而设计的,所以使用了锁,会导致性能不佳。因此在 Python 中可以使用 deque 的 append() 和 popleft() 函数来快速实现队列的功能。
解法二:递归法
考虑递归时,卡在了如何区分第几层。此时只需在传参时,将当前节点所处的层数作为第二个参数即可。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
res = []
if not root:return res
def countlevel(node, level):
if len(res) == level:
res.append([])
res[level].append(node.val)
if node.left:
countlevel(node.left, level+1)
if node.right:
countlevel(node.right, level+1)
countlevel(root, 0)
return res
执行结果:通过显示详情
执行用时 :32 ms, 在所有 python3 提交中击败了98.90%的用户
内存消耗 :13.2 MB, 在所有 python3 提交中击败了94.93%的用户
官解参考