剑指 27 二叉树的镜像

剑指 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]

---

考查知识点

遍历访问二叉树、中序遍历

---

好的解法

这道题乍一看没怎么想出来思路。看了剑指豁然开朗。关于树的题目还是要多画图，先把输入和满足条件的输出在纸上画出来，寻找输入与输出之间的规律，这个规律就是要用代码达到的效果。

关于树的问题，基本上都能从最简单的父节点与左右子节点上找到规律。示例中输入和输出的两棵树，每一个节点的左右子节点都交换了位置，那么这道题其实就是遍历访问每一个树节点，将当前节点的左右子节点交换位置。

既然又是要遍历访问树的每一个节点，自然又是像【剑指 26 树的子结构】一样可以使用递归。这道题如果在本地检查效果的话，免不了要写一个测试函数printInorder来遍历镜像以后树的每一个节点，看看是否镜像对了。这里printInorder是一个中序遍历输出节点值的函数。

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

class Solution {
public:
    TreeNode* mirrorTree(TreeNode* root) {
        if (root == nullptr || (root->left == nullptr && root->right == nullptr))
            return root;

        //交换当前根节点的左右子节点
        //保存原来的右子节点
        TreeNode *oriRight = root->right;
        root->right = root->left;
        root->left = oriRight;

        TreeNode *leftTreeHead  = mirrorTree(root->left);
        TreeNode *rightTreeHead = mirrorTree(root->right);

        return root;
    }

    void printInorder(TreeNode* node)
    {
        if (node == nullptr)
            return;

        cout << node->val << ' ';
        printInorder(node->left);
        printInorder(node->right);

    }
};