第16天|104.二叉树的最大深度、559.n叉树的最大深度、111.二叉树的最小深度、222.完全二叉树的结点个数

104.二叉树的最大深度

思考

什么是深度，什么是高度，如何求深度，如何求高度
原题目：力扣题目104
代码随想录：文章讲解/视频讲解

• 二叉树深度：指从根节点到该节点的最长简单路径边的条数或者节点数（取决于深度从0开始还是从1开始，力扣里一般认为根节点的深度为1）
• 二叉树高度：指从该节点到叶子节点的最长简单路径边的条数或者节点数（取决于高度从0开始还是从1开始）
• 如何求深度：一般使用前序遍历
• 如何求高度：一般使用后序遍历

python代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
    	'''
	    递归三部曲：
	    1）确定递归参数和返回值：节点root，返回深度
	    2）确定终止条件：空节点，返回深度0
	    3）确定单层逻辑：遍历节点的左右节点，返回其中最大的深度 + 1
	    '''
        def recur(root):
            if not root:
                return 0
            left = recur(root.left)
            right = recur(root.right)
            return max(left, right) + 1
        return recur(root)

559.n叉树的最大深度

思考

整体思路和104.二叉树的最大深度保持一致。不同之处在于，要遍历某个节点的所有子节点。

python代码

"""
# Definition for a Node.
class Node:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children
"""

class Solution:
    def maxDepth(self, root: 'Node') -> int:
	    '''
	    递归三部曲：
	    1）确定递归参数和返回值：节点root，返回深度
	    2）确定终止条件：空节点，返回深度0
	    3）确定单层逻辑：遍历节点所有子节点，取其中最大的深度tmp_max，返回tmp_max + 1
	    '''
        if not root: 
            return 0
        tmp_max = 0
        for child in root.children:
            tmp_max = max(tmp_max, self.maxDepth(child))
        return tmp_max + 1

111.二叉树的最小深度

思考

与104.二叉树的最大深度不同之处在于，最小深度不是直接取min(左子树深度, 右子树深度) + 1（如下图所示）。
本题可以使用后序遍历，其实求的是根节点到叶子节点的最小距离，就是求高度的过程，这不过这个最小距离 也同样是最小深度。

【注意】题目中说的是：最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

python代码

class Solution:
    def minDepth(self, root: TreeNode) -> int:
	    '''
	    递归三部曲
	    1）确定递归参数和返回值：传入节点node，返回node的深度
	    2）确定递归终止条件：遇见空节点，返回0
	    3）确定单层递归的逻辑：如果左节点为空且右节点不为空，返回右节点深度+1；如果右节点为空且左节点不为空，返回左节点深度+1；否则，返回左右节点深度的最小值+1
	    '''
	    if not node: #没有节点，
	        return 0
	    if not node.left and node.right:
	        return 1 + minDepth(node.right)
	    if not node.right and node.left:
	        return 1 + minDepth(node.left)
	    return 1 + min(minDepth(node.left), minDepth(node.right))
	return minDepth(root)

222.完全二叉树的结点个数

思考

考虑使用完全二叉树的特性，详见卡尔大神的讲解

python代码

class Solution:
	'''
	普通做法，时间复杂度n 空间logn
	'''
    def countNodes(self, root: TreeNode) -> int:
        return self.getNodesNum(root)
        
    def getNodesNum(self, cur):
        if not cur:
            return 0
        leftNum = self.getNodesNum(cur.left) #左
        rightNum = self.getNodesNum(cur.right) #右
        treeNum = leftNum + rightNum + 1 #中
        return treeNum

class Solution:
	'''
	利用完全二叉树特性，最优解，时间复杂度logn*logn 空间复杂度 logn
	'''
    def countNodes(self, root: TreeNode) -> int:
        if not root:
            return 0
        left = root.left
        right = root.right
        leftDepth = 0 #这里初始为0是有目的的，为了下面求指数方便
        rightDepth = 0
        while left: #求左子树深度
            left = left.left
            leftDepth += 1
        while right: #求右子树深度
            right = right.right
            rightDepth += 1
        if leftDepth == rightDepth:
            return (2 << leftDepth) - 1 #注意(2<<1) 相当于2^2，所以leftDepth初始为0
        return self.countNodes(root.left) + self.countNodes(root.right) + 1

总结

二叉树的题目仍需要多多练习，争取能自己列出来递归三部曲~