【python刷题】二叉树-相关题目

def count(root):
    if not root:
        return 0
   return 1 + count(root.left) + count(root.right)   

def count_depth(root):
    if not root:
        return 0
    return max(count_depth(root.left), count_depth(root.right)) + 1

class Solution:
    def invertTree(self, root: TreeNode) -> TreeNode:
        if not root:
            return None
        root.left, root.right = root.right, root.left
        self.invertTree(root.left)
        self.invertTree(root.right)
        return root

class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        if not root:
            return None
        self.flatten(root.left)
        self.flatten(root.right)
        left = root.left
        right = root.right
        root.left = None
        root.right = left
        p = root
        while p.right:
            p = p.right
            
        p.right = right
        return root