剑指Offer27.二叉树的镜像Golang版

最新推荐文章于 2022-05-25 21:25:46 发布

原创最新推荐文章于 2022-05-25 21:25:46 发布 · 181 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树 #算法 #数据结构 #dfs

leetcode刷题专栏收录该内容

252 篇文章

订阅专栏

本文介绍如何使用Golang实现二叉树的镜像转换。通过两种不同的递归方法，即深度优先遍历和后续遍历，来实现二叉树节点的左右子树互换，最终达到镜像的效果。代码示例清晰地展示了每种方法的具体实现过程。

剑指Offer27.二叉树的镜像Golang版

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

2. 思路

2.1. 思路1

二叉树深度优先遍历，遍历过程中进行左右子树交换操作。

二叉树的深度优先遍历，也就是先序遍历

2.2. 思路2

二叉树先进行递归遍历，找到叶子节点，回溯的过程中进行左右子树交换操作。

或称为后续遍历

3. 代码

3.1. 思路1代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func mirrorTree(root *TreeNode) *TreeNode {
    p := root
    dfs(p)
    return root
}

func dfs(root *TreeNode) {
    if root == nil {
        return
    }
	
	// 向下递归遍历，过程中进行交换
    root.Left, root.Right = root.Right, root.Left

    dfs(root.Left)
    dfs(root.Right)
}

3.2. 思路2代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func mirrorTree(root *TreeNode) *TreeNode {
   if root == nil {
       return root
   }
   
   // 先递归遍历，找到最下面的叶子节点
   left := mirrorTree(root.Left)
   right := mirrorTree(root.Right)
   // 回溯的过程中，从下向上进行交换赋值
   root.Left = right
   root.Right = left
   
   return root
}