complete invertTree
This commit is contained in:
parent
b71f6978f1
commit
9e84d50bb5
1 changed files with 5 additions and 5 deletions
|
|
@ -37,11 +37,11 @@ enum LeetCode {
|
|||
}
|
||||
|
||||
static func invertTree(_ root: TreeNode?) -> TreeNode? {
|
||||
var invertedTree: TreeNode
|
||||
if root?.left != nil || root?.right != nil {
|
||||
guard let nodeVal = root?.val else {
|
||||
}
|
||||
invertedTree = TreeNode(nodeVal, root?.right, root?.left)
|
||||
var invertedTree: TreeNode?
|
||||
if let val = root?.val {
|
||||
invertedTree = TreeNode(val, invertTree(root?.right), invertTree(root?.left))
|
||||
} else {
|
||||
invertedTree = root
|
||||
}
|
||||
return invertedTree
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue