树的镜像&栈模拟递归

这篇文章有两个点,第一个是题目本身,第二个是我觉得很不错的用栈来模拟递归。

题目

做过这么多树的题目后应该觉得用递归很简单吧,我们只需要交换每一个节点的左右节点就行了。
要注意这里要交换指针才能达到效果,交换数是不行的。

然后简单的递归就实现了:

void MirrorReverseTree(treeNode* tree)
{
    if (!tree)
        return;
    //swap left&right
    treeNode* tmp = tree->left;
    tree->left = tree->right;
    tree->right = tmp;

    if (tree->left)
        MirrorReverseTree(tree->left);
    if (tree->right)
        MirrorReverseTree(tree->right);
}

用循环来实现

有人说所有的递归都可以用循环来实现。。但是可以肯定的是,使用递归的时候编写代码要简单一些。
但是使用递归的话,要小心,避免stackoverflow。

这个题呢,我们要遍历每个节点,然后做和递归同样的事情,可以用while循环判断是否到达了叶子节点,可是我仍然觉得直接这样写挺难的,我再想想。。

然而有另一种很巧妙的思路:

由于递归的本质是编译器生成了一个函数调用的栈,因此用循环来完成同样任务时最简单的办法就是用一个辅助栈来模拟递归。首先我们把树的头结点放入栈中。在循环中,只要栈不为空,弹出栈的栈顶结点,交换它的左右子树。 如果它有左子树, 把它的左子树压入栈中; 如果它有右子树, 把它的右子树压入栈中。这样在下次循环中就能交换它儿子结点的左右子树了。

这种认识是很重要的。

所以我们利用这种想法来实现,我们每经过一个节点就把这个节点压入栈,然后对每个节点做同样的事情就可以了。

其实这种题就是说,怎么简单的遍历到整个树。

code:

//non recursive way:
//using a stack
void mirrorReverseTree_2(treeNode* tree)
{
    if (!tree)
        return;
    std::stack<treeNode*> nodeStack;
    nodeStack.push(tree);
    while (nodeStack.size())
    {
        //swap left&right
        treeNode* tmp = tree->left;
        tree->left = tree->right;
        tree->right = tmp;

        if (tree->left)
            nodeStack.push(tree->left);
        if (tree->right)
            nodeStack.push(tree->right);
    }
    
}

其实这个函数里,
if (tree->left) nodeStack.push(tree->left); if (tree->right) nodeStack.push(tree->right);
这两个语句让人深深的感觉到有递归的影子啊!

文章参考何海涛大神文章

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • B树的定义 一棵m阶的B树满足下列条件: 树中每个结点至多有m个孩子。 除根结点和叶子结点外,其它每个结点至少有m...
    文档随手记阅读 13,470评论 0 25
  • 课程介绍 先修课:概率统计,程序设计实习,集合论与图论 后续课:算法分析与设计,编译原理,操作系统,数据库概论,人...
    ShellyWhen阅读 2,388评论 0 3
  • 一直以来,我都很少使用也避免使用到树和图,总觉得它们神秘而又复杂,但是树在一些运算和查找中也不可避免的要使用到,那...
    24K男阅读 6,794评论 5 14
  • 第一章 绪论 什么是数据结构? 数据结构的定义:数据结构是相互之间存在一种或多种特定关系的数据元素的集合。 第二章...
    SeanCheney阅读 5,867评论 0 19
  • 那留在时光里的 不论峥嵘,抑或平庸 都将尘封 未来的一段征程 蕴藏无限可能 等待释放 夜空中北斗依旧 不迷惘,星夜...
    月圆天心阅读 208评论 0 0