mirror of
https://github.com/autistic-symposium/master-algorithms-py.git
synced 2025-05-24 01:11:35 -04:00
Update and rename tree_bfs.py to bt_bfs.py
This commit is contained in:
parent
9d1aa92451
commit
070071e687
1 changed files with 1 additions and 4 deletions
|
@ -1,34 +0,0 @@
|
|||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
# author: bt3gl
|
||||
|
||||
# Given the root of a binary tree, return the level order traversal of its nodes' values.
|
||||
# (i.e., from left to right, level by level).
|
||||
|
||||
|
||||
def level_order(root: Optional[TreeNode]) -> list[list[int]]:
|
||||
|
||||
if root is None:
|
||||
return []
|
||||
|
||||
queue = collections.deque()
|
||||
queue.append(root)
|
||||
result = []
|
||||
|
||||
while queue:
|
||||
|
||||
this_level = []
|
||||
|
||||
for _ in range(len(queue)):
|
||||
|
||||
current = queue.popleft()
|
||||
|
||||
if current:
|
||||
this_level.append(current.val)
|
||||
queue.append(current.left)
|
||||
queue.append(current.right)
|
||||
|
||||
if this_level:
|
||||
result.append(this_level)
|
||||
|
||||
return result
|
Loading…
Add table
Add a link
Reference in a new issue