剑指offer——二叉树的镜像(递归实现已解决,循环实现待解决)

题目描述

操作给定的二叉树,将其变换为源二叉树的镜像。

输入描述:

二叉树的镜像定义:源二叉树 
    	    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,这样更简洁。

 

书上代码:

剑指offer——二叉树的镜像(递归实现已解决,循环实现待解决)_第1张图片

思考

上面的代码是用递归实现的,如果要求用循环,该如何实现?

 

你可能感兴趣的:(C++,剑指offer)