247 字
1 分钟
二叉树的镜像

题目描述#

请完成一个函数,输入一个二叉树,该函数输出它的镜像。

例如输入:

     4
   /   \
  2     7
 / \   / \
1   3 6   9

镜像输出:

     4
   /   \
  7     2
 / \   / \
9   6 3   1

示例 1:

输入:root = [4,2,7,1,3,6,9] 输出:[4,7,2,9,6,3,1]  

限制:

0 <= 节点个数 <= 1000

注意:本题与主站 226 题相同:https://leetcode-cn.com/problems/invert-binary-tree/

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/er-cha-shu-de-jing-xiang-lcof 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解一#

递归

func mirrorTree(_ root: TreeNode?) -> TreeNode? {
if root == nil {
return nil
}
let left = root?.left
let right = root?.right
root?.left = mirrorTree(right)
root?.right = mirrorTree(left)
return root
}

题解二#

stack

func mirrorTree2(_ root: TreeNode?) -> TreeNode? {
var stack: [TreeNode?] = [root]
while stack.last != nil {
let node = stack.removeLast()
let right = node?.right
let left = node?.left
if let rn = right {
stack.append(rn)
}
if let ln = left {
stack.append(ln)
}
node?.left = right
node?.right = left
}
return root
}