【LeetCode】剑指 Offer 27. 二叉树的镜像 - Go 语言题解

---

一、题目描述

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

例如输入：

     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

---

二、我的题解 - 递归

从根节点起，从顶至底，交换每个节点的左右孩子，直到越过叶子节点。

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func mirrorTree(root *TreeNode) *TreeNode {
    if root == nil{
        return nil
    }else{
        return symm(root)
    }
}

func symm(root *TreeNode) *TreeNode{
    var temp1,temp2 *TreeNode
    if root == nil{
        return nil
    }else if root.Left == nil && root.Right == nil {
        return root
    }else{
        temp1 = symm(root.Left)
        temp2 = symm(root.Right)
        root.Left = temp2
        root.Right = temp1
        return root
    }

}

评判结果：

实际上，你会发现上述代码是有重复部分的，可以改进：

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func mirrorTree(root *TreeNode) *TreeNode {
    var temp1,temp2 *TreeNode
    if root == nil{
        return nil
    }else if root.Left == nil && root.Right == nil {
        return root
    }else{
        temp1 = mirrorTree(root.Left)
        temp2 = mirrorTree(root.Right)
        root.Left = temp2
        root.Right = temp1
        return root
    }
}