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? {
|
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
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue