本次全部使用递归的方式实现
104.二叉树的最大深度
递归法,后序遍历
class Solution:
def maxDepth(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
left_depth = self.maxDepth(root.left)
right_depth = self.maxDepth(root.right)
return 1 + max(left_depth, right_depth)
111.二叉树的最小深度
注意:最小深度是从根节点到最近叶子节点的最短路径上的节点数量。注意是叶子节点,左右孩子都为空的节点才是叶子节点。因此终止条件多了几个判断。
class Solution:
def minDepth(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
left_depth = self.minDepth(root.left)
right_depth = self.minDepth(root.right)
if not root.left:
return 1 + right_depth
if not root.right:
return 1 + left_depth
return 1 + min(left_depth, right_depth)
222.完全二叉树的节点个数
普通二叉树的递归方式
class Solution:
def countNodes(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
left_nodes = self.countNodes(root.left)
right_nodes = self.countNodes(root.right)
return 1 + left_nodes + right_nodes
完全二叉树的递归方式
class Solution:
def countNodes(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
left = root.left
right = root.right
left_depth = 0
right_depth = 0
while left:
left = left.left
left_depth += 1
while right:
right = right.right
right_depth += 1
if left_depth == right_depth:
return (2 << left_depth) - 1
return 1 + self.countNodes(root.left) + self.countNodes(root.right)