算法训练营Day16(二叉树3)

104.二叉树的最大深度力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

思考

什么是深度，什么是高度，如何求深度（使用前序遍历），如何求高度（使用后序遍历），这里有关系到二叉树的遍历方式。

• 二叉树节点的深度：指从根节点到该节点的最长简单路径边的条数或者节点数（取决于深度从0开始还是从1开始）
• 二叉树节点的高度：指从该节点到叶子节点的最长简单路径边的条数或者节点数（取决于高度从0开始还是从1开始）

# 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:
        return self.getdepth(root)
        
    def getdepth(self, node):
        if not node:
            return 0
        leftheight = self.getdepth(node.left) #左
        rightheight = self.getdepth(node.right) #右
        height = 1 + max(leftheight, rightheight) #中
        return height

111.二叉树的最小深度 力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

提醒

和最大深度 看似差不多，其实 差距还挺大，有坑。

class Solution:
    def minDepth(self, root):
        return self.getDepth(root)
    def getDepth(self, node):
        if node is None:
            return 0
        leftDepth = self.getDepth(node.left)  # 左
        rightDepth = self.getDepth(node.right)  # 右
        
        # 当一个左子树为空，右不为空，这时并不是最低点
        if node.left is None and node.right is not None:
            return 1 + rightDepth
        
        # 当一个右子树为空，左不为空，这时并不是最低点
        elif node.left is not None and node.right is None:
            return 1 + leftDepth
        
        else:
            result = 1 + min(leftDepth, rightDepth)
            return result

222.完全二叉树的节点个数力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

 提醒

需要了解普通二叉树 怎么求，完全二叉树又怎么求

第一种（普通二叉树解法）

class Solution:
    def countNodes(self, root: Optional[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:
    def countNodes(self, root: TreeNode) -> int:
        if not root:
            return 0
        return 1 + self.countNodes(root.left) + self.countNodes(root.right)

第二种（完全二叉树解法）

class Solution:
    def countNodes(self, root: Optional[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

        leftNum = self.countNodes(root.left) #左
        rightNum = self.countNodes(root.right) #右
        treeNum = leftNum + rightNum + 1 #中
        return treeNum

精简版

class Solution: # 利用完全二叉树特性
    def countNodes(self, root: TreeNode) -> int:
        if not root: return 0
        count = 1
        left = root.left; right = root.right
        while left and right:
            count+=1
            left = left.left; right = right.right
        if not left and not right: # 如果同时到底说明是满二叉树，反之则不是
            return 2**count-1
        return 1+self.countNodes(root.left)+self.countNodes(root.right)