题目
https://leetcode-cn.com/problems/maximum-width-of-binary-tree/
https://leetcode-cn.com/problems/er-cha-shu-de-shen-du-lcof/
深度——DFS
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if not root:
return 0
return max(self.maxDepth(root.left), self.maxDepth(root.right))+1
当前深度为左右子树最大深度+1
宽度——层序遍历
想法
宽度即按层来计算,所以采用层序遍历,所注意的一点是:普通层序遍历在对列pop一点,随机可以加入两个子节点,但是在本题中各层应该分开储存,以便于计算每层的节点数(即宽度)
关于宽度,采用二叉树节点在数组中的位置,即某节点为i,则左孩子为i*2,右孩子为i*2+1,父节点为 int(i/2)
class Solution:
def widthOfBinaryTree(self, root: TreeNode) -> int:
if not root:
return 0
ans = 1
queue = [(0, root)] # 起始坐标,节点
while queue:
ans = max(ans, queue[-1][0] - queue[0][0] + 1) # 每次循环都是一层,计算上一层宽度
temp = [] # 临时数组,实现每层的替换
for i, node in queue:
if node.left: # 子树为空,不加入
temp.append((2*i, node.left))
if node.right:
temp.append((2*i+1, node.right))
queue = temp
return ans