文章目录
一、题目描述
请完成一个函数,输入一个二叉树,该函数输出它的镜像。
例如输入:
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
}
}