complete invertTree

This commit is contained in:
cediackermann 2026-06-15 08:36:20 +02:00
parent b71f6978f1
commit 9e84d50bb5
No known key found for this signature in database

View file

@ -37,11 +37,11 @@ enum LeetCode {
} }
static func invertTree(_ root: TreeNode?) -> TreeNode? { static func invertTree(_ root: TreeNode?) -> TreeNode? {
var invertedTree: TreeNode var invertedTree: TreeNode?
if root?.left != nil || root?.right != nil { if let val = root?.val {
guard let nodeVal = root?.val else { invertedTree = TreeNode(val, invertTree(root?.right), invertTree(root?.left))
} } else {
invertedTree = TreeNode(nodeVal, root?.right, root?.left) invertedTree = root
} }
return invertedTree return invertedTree
} }