Update bt_is_tree_symmetric.py

This commit is contained in:
bt3gl 2023-08-08 13:56:19 -07:00 committed by GitHub
parent 153e85ca03
commit f41011e849
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,7 +3,7 @@
# author: bt3gl
def is_symmetric(root: Optional[Node]) -> bool:
def is_symmetric(root) -> bool:
stack = [(root, root)]
@ -14,21 +14,21 @@ def is_symmetric(root: Optional[Node]) -> bool:
if (not node1 and node2) or (not node2 and node1):
return False
elif node1 and node2 and node1.val != node2.val:
return False
elif not node1 and not node2:
continue
elif node1 and node2 and node1.val != node2.val:
return False
stack.append([node1.left, node2.right])
stack.append([node1.right, node2.left])
return True
def is_symmetric_recursive(root: Optional[TreeNode]) -> bool:
def is_symmetric_recursive(root) -> bool:
def helper(node1, node2):
def helper(node1, node2):
if (not node1 and node2) or \
(not node2 and node1) or \
(node1 and node2 and node1.val != node2.val):
@ -39,5 +39,5 @@ def is_symmetric_recursive(root: Optional[TreeNode]) -> bool:
return helper(node1.left, node2.right) and helper(node2.left, node1.right)
return helper(root.left, root.right)
return helper(root.left, root.right)