二叉树的最大宽度python
时间: 2023-06-29 10:18:40 浏览: 108
Python3实现二叉树的最大深度
二叉树的宽度是指各层节点数的最大值,最大宽度即为二叉树宽度的最大值。下面是一个 Python 实现的二叉树最大宽度的代码示例:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def widthOfBinaryTree(root: TreeNode) -> int:
if not root:
return 0
queue = [(root, 1)]
max_width = 0
while queue:
level_size = len(queue)
_, level_head_index = queue[0]
for i in range(level_size):
node, col_index = queue.pop(0)
if node.left:
queue.append((node.left, 2 * col_index))
if node.right:
queue.append((node.right, 2 * col_index + 1))
max_width = max(max_width, col_index - level_head_index + 1)
return max_width
```
该算法使用队列进行二叉树的层次遍历,并记录每个节点的列索引。在遍历每一层时,计算该层的宽度,并更新最大宽度。最终返回二叉树的最大宽度即可。
阅读全文