剑指 Offer 27. 二叉树的镜像

听钱塘信起

于 2022-10-10 11:23:39 发布

阅读量65

点赞数 2

分类专栏：《剑指offer》(第2版) 文章标签：算法

本文链接：https://blog.csdn.net/chenyunjiangNN/article/details/127239613

版权

《剑指offer》(第2版) 专栏收录该内容

40 篇文章 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

二、递归写法

只要结点存在叶子结点、便计算其镜像后交换位置；

时间复杂度O(N)、N为二叉树的结点树。空间复杂度O(N),N为二叉树的深度。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode mirrorTree(TreeNode root) {
        if(root ==null){
            return null;
        }
        TreeNode temp = root.left;
        root.left=mirrorTree(root.right);
        root.right = mirrorTree(temp);
        return root;
    }
}

执行结果

三、栈+深度优先搜索(DFS)

class Solution {
    public TreeNode mirrorTree(TreeNode root) {
       if(root == null) {
            return null;
       }
        Stack<TreeNode> stack = new Stack<>();
        // 根结点入栈
        stack.push(root);
    
        while(!stack.isEmpty()) {
            // 移除栈顶元素
            TreeNode node = stack.pop();
            //如果当前节点的左子树不为空，就把左子树节点入栈
            if(node.left != null){
                stack.push(node.left);
            }
             //如果当前节点的右子树不为空，就把右子树节点入栈
            if(node.right != null){
                stack.push(node.right);
            } 

        // 交换node的两个叶子结点
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;
        }

        return root;
    }
}

执行结果

四、队列+广度优先搜索(BFS)

 
class Solution {
    public TreeNode mirrorTree(TreeNode root) {

        if (root == null){
             return null;
        }
          
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            //出队
            TreeNode node = queue.poll();
            //交换node节点的两个子节点
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;
            //如果当前节点的左子树不为空，就把左子树节点加入到队列中
            if (node.left != null) {
                queue.add(node.left);
            }
            //如果当前节点的右子树不为空，就把右子树节点加入到队列中
            if (node.right != null) {
                queue.add(node.right);
            }
        }
        return root;
    }
}

执行结果：