二叉树的镜像(递归法、辅助栈法)

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

 

例如输入：

     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

 

解题思路：(递归法)

二叉树的任意结点，设其左孩子为left，右孩子为right，则其镜像二叉树的左孩子为right，右孩子为left。可以从根结点开始，存储其左孩子left，然后使用递归交换其左右孩子。

 

算法流程：

1.如果该结点为空，返回空(递归终止条件)

2.创建结点tmp存储该结点的左孩子left

3.递归该结点的右孩子right并把其值赋给该结点的左孩子left

4.递归该结点的左孩子tmp并把其值赋给该结点的右孩子right

 

代码：

class Solution {

public:

    TreeNode* mirrorTree(TreeNode* root) {

    if(root==nullptr)

        return nullptr;

    TreeNode* tmp=root->left;

   root->left=mirrorTree(root->right);

   root->right=mirrorTree(tmp);

    return root;

    }

};

 

运行步骤：

例如输入：

     4

   / \

  2 7

 / \ / \

1 3 6 9

 

tmp=2，4->left=mirrorTree(7);

tmp=6，7->left=mirrorTree(9);

遇到终止条件，返回9，7->left=9;

7->right=mirrorTree(6);

遇到终止条件，返回6，7->right=6;

函数返回7，4->left=7;

4->right=mirrorTree(2);

tmp=1，2->left=mirrorTree(3);

遇到终止条件，返回3，2->left=3;

7->right=mirrorTree(1);

遇到终止条件，返回1，2->right=1;

函数返回2，4->right=2;

函数返回4 (镜像二叉树根结点)

 

镜像输出：

     4

   / \

  7 2

 / \ / \

9 6 3 1

 

解题思路：辅助栈法

利用栈遍历树的所有结点，并交换每个结点的左右孩子。

 

算法流程：

1.判断是否是空树，是空树则返回空

2.创建栈st，并把树的根结点root入栈

3.循环(栈不为空)

               创建结点node存储栈顶元素

               栈顶元素出栈

               如果node的左孩子不为空

                       node的左孩子入栈

               如果node的右孩子不为空

                       node的右孩子入栈

               交换node的左右孩子

4.返回根结点root

 

代码：

class Solution {

public:

    TreeNode* mirrorTree(TreeNode* root) {

    if(root==nullptr) return nullptr;

    stack<TreeNode*> st;

    st.push(root);

    while(!st.empty()) {

        TreeNode* node=st.top();

        st.pop();

        if(node->left!=nullptr)

            st.push(node->left);

        if(node->right!=nullptr)

            st.push(node->right);

        TreeNode* tmp=node->left;

        node->left=node->right;

        node->right=tmp;

    }

    return root;

    }

};

 

运行步骤：

例如输入：

     4

   / \

  2 7

 / \ / \

1 3 6 9

栈：[4]  4出栈，2，7入栈，交换2和7

    4

   / \

  7  2

 / \ / \

6 9 1 3

栈：[2，7]  7出栈，6，9入栈，交换6和9

     4

   / \

  7  2

 / \ / \

9 6 1 3

栈：[2，6，9]  9出栈，6出栈

栈：[2]  2出栈，1，3入栈，交换1和3

     4

   / \

  7  2

 / \ / \

9 6 3 1

栈：[1，3]  3出栈，1出栈

返回根结点4