Delete Leaves With a Given Value LeetCode solution in Python

1
class Solution:
2
    def removeLeafNodes(
3
        self, root: Optional[TreeNode], target: int
4
    ) -> Optional[TreeNode]:
5
        if not root:
6
            return None
7
        root.left = self.removeLeafNodes(root.left, target)
8
        root.right = self.removeLeafNodes(root.right, target)
9
        if not root.left and not root.right and root.val == target:
10
            return None
11
        return root