请完成一个函数,输入一个二叉树,该函数输出它的镜像。
例如输入:
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