Time: 43 ms (75.33%), Space: 17.1 MB (80.39%) - LeetHub

Author: hogan-tech

1325-delete-leaves-with-a-given-value/1325-delete-leaves-with-a-given-value.py

@@ -0,0 +1,29 @@
+from typing import Optional
+
+
+class TreeNode:
+    def __init__(self, val=0, left=None, right=None):
+        self.val = val
+        self.left = left
+        self.right = right
+
+
+class Solution:
+    def removeLeafNodes(self, root: Optional[TreeNode], target: int) -> Optional[TreeNode]:
+        if not root :
+            return None
+        root.left = self.removeLeafNodes(root.left, target)
+        root.right = self.removeLeafNodes(root.right, target)
+        if not root.left and not root.right and root.val == target:
+            return None
+        return root
+
+
+root = TreeNode(1)
+root.left = TreeNode(2)
+root.left.left = TreeNode(2)
+root.right = TreeNode(3)
+root.right.left = TreeNode(2)
+root.right.right = TreeNode(4)
+target = 2
+print(Solution().removeLeafNodes(root, target))