104.二叉树的最大深度
最大深度这个题目比较简单,如果使用递归方法,其实如果只有三个节点,就是本节点加上左右两个节点去比较。那么左右不是一个节点也一样,就是比较两边之和。递归终止条件是如果遇到None,那么直接返回0就好了。每层逻辑是当前节点数量1,然后左右比较。那么题解如下:
class Solution:
def maxDepth(self, root: Optional[TreeNode]) -> int:
def helper(root, depth):
if not root:
return depth
return max(helper(root.left, depth + 1), helper(root.right, depth + 1))
return helper(root, 0)
111.二叉树的最小深度
此题看着和最大深度完全一样,其实有细小差别。方法还是那个方法,就是比较左右最小的。但是问题在于,如果有一边没有节点,是不算的,不能记作0。因此判断条件要单独写出来有一边为None的情况,让它不计算数量就好了。题解如下:
class Solution:
def minDepth(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
if not root.left:
return self.minDepth(root.right) + 1
if not root.right:
return self.minDepth(root.left) + 1
return min(self.minDepth(root.left), self.minDepth(root.right)) + 1
222.完全二叉树的节点个数
这道题如果是普通二叉树,那么方法特别简单,直接遍历求数量也好,还是用一个递归统计也好,都可以直接写出来。不过本题是完全二叉树,那么我们只需要找到里面所有的满树,然后用公式算个数,这样会比每个都遍历快很多。题解如下:
class Solution:
def countNodes(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
return 1 + self.countNodes(root.left) + self.countNodes(root.right)