镜像二叉树即交换原二叉树的左右子树,可以使用递归,输入函数的每一个节点,先判断该节点是否为空,为空则返回nullptr,说明空二叉树或已经遍历到叶子节点下左右节点为空,否则判断该节点的左右节点只要有一个不为空,则交换左右节点。
代码如下:
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return TreeNode类
*/
TreeNode* Mirror(TreeNode* pRoot) {
// write code here
if(pRoot==nullptr) return nullptr;
if(pRoot->left!=nullptr||pRoot->right!=nullptr)
{
TreeNode *temp=pRoot->left;
pRoot->left=pRoot->right;
pRoot->right=temp;
}
Mirror(pRoot->left);
Mirror(pRoot->right);
return pRoot;
}
};