题解 | #二叉树的镜像#
二叉树的镜像
https://www.nowcoder.com/practice/a9d0ecbacef9410ca97463e4a5c83be7
二叉树的镜像:最直观的想法是,交换每一个结点的左右孩子结点。首先编写一个函数toMirror,用于交换每一个结点的左右孩子结点,其参数为TreeNode类型指针,返回值为void,当当前结点指针为空时返回,否则使用swap交换左右子树,然后先对当前结点的左孩子使用toMirror,再对当前结点的右孩子使用toMirror。在给定函数Mirror中,首先判断根结点是否为空,如果是则返回nullptr,否则调用toMirror,然后再返回根结点。
//交换左右子树
void toMirror(TreeNode * cur)
{
if(!cur) //当前结点为空则返回
return;
swap(cur->left,cur->right); //交换左右子树
toMirror(cur->left); //左子树镜像
toMirror(cur->right); //右子树镜像
}
TreeNode* Mirror(TreeNode* pRoot)
{
if(!pRoot)
return nullptr;
toMirror(pRoot);
return pRoot;
}
优化:对比上述边界条件以及函数头可知,可以将两者进行结合,从而变成一个函数。曾经我对递归一直有疑问,尤其是二叉树这一部分,到底是一个函数还是两个函数,到底是前序遍历、中序遍历还是后序遍历……但其实对于二叉树以及递归不是很熟悉的话,可以先按照自己的解题思路来,再将解题思路慢慢转化为实现代码,再进行一步步的优化,练习多了自然就会有经验啦。
TreeNode* Mirror(TreeNode* pRoot)
{
if(!pRoot) //当前结点为空则返回
return nullptr;
swap(pRoot->left,pRoot->right); //交换左右子树
Mirror(pRoot->left); //左子树镜像
Mirror(pRoot->right); //右子树镜像
return pRoot; //返回根结点
}
#二叉树的镜像#剑指offer 文章被收录于专栏
剑指offer专栏主要分享剑指offer题解。
查看22道真题和解析