题目描述
操作给定的二叉树,将其变换为源二叉树的镜像。
输入描述:
二叉树的镜像定义:源二叉树
8
/ \
6 10
/ \ / \
5 7 9 11
镜像二叉树
8
/ \
10 6
/ \ / \
11 9 7 5
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {}
};
*/
/*
先序遍历每棵树的每个节点,如果遍历到的节点有子节点,就交换他的两个子节点。
当交换完所有非叶子节点的左右子节点后,得到树的镜像。
*/
class Solution
{
public:
void Mirror(TreeNode* pRoot)
{
TreeNode* pNode = pRoot;
if (pNode == nullptr) //根节点为空,返回
return;
if (pNode->left == nullptr && pNode->right == nullptr) //遍历完非叶节点
return;
TreeNode* pTemp = pNode->left; //左子节点与右子节点交换
pNode->left = pNode->right;
pNode->right = pTemp;
if (pNode->left) //遍历当前节点的左子树
Mirror(pNode->left);
if(pNode->right) //遍历当前节点的右子树
Mirror(pNode->right);
}
};
简单写法
class Solution
{
public:
void Mirror(TreeNode *pRoot)
{
if(pRoot== nullptr)
return;
swap(pRoot->left,pRoot->right);
Mirror(pRoot->left);
Mirror(pRoot->right);
return;
}
};