剑指 Offer 27. 二叉树的镜像

难度简单

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

例如输入:

     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

解题思路
通过题目我们发现,二叉树的镜像就是通过交换原二叉树中所有结点的左子树和右子树变换而成的,因此我们就需要编码来实现这种交换过程。

递归
模型:二叉树的先序遍历(遍历二叉树的所有结点)
递归返回条件:当前结点为 NULL
实现操作:交换根结点的左右子树
代码

class Solution {
public:
    TreeNode* mirrorTree(TreeNode* root) {
        if (root == NULL) {
            return NULL;
        }
        swap(root->left, root->right);
        mirrorTree(root->left);
        mirrorTree(root->right);
        return root;
    }
};
栈模拟
模型:栈模拟二叉树的先序遍历
循环结束条件:栈为空
实现操作:交换栈顶结点的左右子树
代码

class Solution {
public:
    TreeNode* mirrorTree(TreeNode* root) {
        stack s;
        s.push(root);
        while (!s.empty()) {
            TreeNode* node = s.top();
            s.pop();
            if (node == NULL) {
                continue;
            }
            swap(node->left, node->right);
            s.push(node->left); 
            s.push(node->right);
        }
        return root;
    }
};
队列模拟
模型:使用队列模拟二叉树的层次遍历
循环结束条件:队列为空
实现操作:交换队首结点的左右子树
代码

class Solution {
public:
    TreeNode* mirrorTree(TreeNode* root) {
        queue q;
        q.push(root);
        while (!q.empty()) {
            TreeNode* node = q.front();
            q.pop();
            if (node == NULL) {
                continue;
            }
            swap(node->left, node->right);
            q.push(node->left);
            q.push(node->right);
        }
        return root;
    }
};

作者:huwt
链接:https://leetcode-cn.com/problems/er-cha-shu-de-jing-xiang-lcof/solution/er-cha-shu-de-jing-xiang-di-gui-zhan-mo-ni-dui-lie/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

 

你可能感兴趣的:(剑指offer)