题目描述
操作给定的二叉树,将其变换为源二叉树的镜像。
输入描述:
二叉树的镜像定义:源二叉树
8
/ \
6 10
/ \ / \
5 7 9 11
镜像二叉树
8
/ \
10 6
/ \ / \
11 9 7 5
交换所有非叶子节点的左右节点。
class Solution {
public:
void Mirror(TreeNode *pRoot) {
TreeNode *node = pRoot, *temp = NULL;
if(pRoot == NULL) return;
if(node->left != NULL && node->right != NULL)
{
temp = node->left;
node->left = node->right;
node->right = temp;
}
else if(node->left != NULL)
{
node->right = node->left;
node->left = NULL;
}
else if(node->right != NULL)
{
node->left = node->right;
node->right = NULL;
}
else
return;
// if(node->left != NULL)
Mirror(node->left);
// if(node->right != NULL)
Mirror(node->right);
}
};
调错总结:
(1)开始报上面的错误,对照书上代码发现漏掉一个边界条件pRoot == NULL ,可能我运行的第一个测试用例是空树,尝试引用空指针,故报错。另外,左右子树为空(叶子节点,最后一个else)也可以归到这里,直接返回。
(2)最后递归调用左右子树时,也可以不用if判断条件,因为上面会检查传入参数是否为空。
(3)检查节点左右子树是否为空时,都不为空,左子树为空,右子树为空三种需要交换的情况可以归到一起(都用temp),不用temp,这样更简洁。
书上代码:
思考
上面的代码是用递归实现的,如果要求用循环,该如何实现?