题解 | #按之字形顺序打印二叉树#

按之字形顺序打印二叉树

http://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

题目的主要信息:

给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)。

方法一:

层次遍历。用队列进行层次遍历,首先将根结点入队。通过一个循环,当队列不为空时,将队列中的元素逐个出列记录在ans中,同时每出列一个结点,就把该结点的左右孩子入列(如果有的话),这样能保证按照层次对树进行遍历,但是题目要求的是按照之字形顺序打印,实际上就是在偶数层翻转一下,因此用level记录层次,每当到达偶数层就翻转该层的结点。

具体做法:

class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int>> res;//保存遍历结果
        if (pRoot==NULL){//如果根为NULL
            return res;
        }
        queue<TreeNode*> q;
        q.push(pRoot);//根结点入列
        int level = 0;//层
        while (!q.empty()) {//层次遍历
            int len = q.size();
            vector<int> ans;//每一层的
            for(int i = 0;i < len; i++) {
                TreeNode *node = q.front();
                q.pop();
                ans.push_back(node->val);
                if (node->left) q.push(node->left);//左孩子入列
                if (node->right) q.push(node->right);//右孩子入列
            }
            level++;
            if (level%2==0){ //偶数层是从右往左遍历
                reverse(ans.begin(), ans.end());//将该层的值翻转
            }
            res.push_back(ans);
        }
        return res;
    }
    
};

复杂度分析:

  • 时间复杂度:O(n)O(n),需要遍历一遍整个树。
  • 空间复杂度:O(n)O(n),队列大小为n。

方法二:

双栈法。首先明确栈的性质,存入栈中的顺序与从栈中输出的顺序相反。我们可以利用栈的这个性质进行交叉叫顺打印,用stack1保存奇数层的结点,stack2保存偶数层的结点。先把根结点放入stack1中,如果当前stack1不为空,说明当前层是奇数层,逐个输出栈顶元素,将输出结点的子女按照从左到右的顺序存入stack2中,这样能保证到达偶数层时,stack2中的出栈顺序为从右往左;当前层是偶数层,逐个输出栈顶元素,将输出结点的子女按照从右往左的顺序存入stack1中,这样能保证到达奇数层时,stack1中的出栈顺序为从左往右。将每层元素存入result中,最后返回result。 alt 具体做法:

class Solution{
public:
    vector<vector<int> > Print( TreeNode* pRoot){
        vector<vector<int> > result;
        if(pRoot==NULL){//根结点为NULL,返回空
            return result;
        }
        stack<TreeNode*> stack1, stack2;
        stack1.push(pRoot);//放入根结点
        while(!stack1.empty() || !stack2.empty() ){
            vector<int> ret1,ret2;
            TreeNode* cur = NULL;
            while( !stack1.empty()){//奇数层
                cur = stack1.top();
                //从左往右存子女到stack2中,stack2中出栈将以从右往左的顺序
                if(cur->left) stack2.push(cur->left);
                if(cur->right) stack2.push(cur->right);
                ret1.push_back(stack1.top()->val);//保存本层所有结点值
                stack1.pop();
            }
            if(ret1.size()) result.push_back(ret1);
            while( !stack2.empty()){//偶数层
                cur = stack2.top();
                 //从右往左存子女到stack1中,stack1中出栈将以从左往右的顺序
                if(cur->right) stack1.push( cur->right);
                if(cur->left) stack1.push( cur->left);
                ret2.push_back(stack2.top()->val); //保存本层所有结点值
                stack2.pop();
            }
            if(ret2.size()) result.push_back(ret2);
        }
        return result;
    }
};

复杂度分析:

  • 时间复杂度:O(n)O(n),需要遍历一遍整个树。
  • 空间复杂度:O(n)O(n),栈大小为n。
全部评论

相关推荐

头像
01-12 14:44
已编辑
百度_高级研发工程师
今天看到了某平台攻击牛友的帖子,段段今天打算为牛友们说句话,我们的努力到底有没有意义。&nbsp;(原文复述:感觉牛客就是当年那群做题区毕业了开始找工作还收不住那股味,颇有一种从年级第一掉到年纪第二后抱怨考不上大学的区味)&nbsp;&nbsp;粗鄙,无礼,傲慢,攻击,在这里我没有看到任何有用的分析,我只看到了屁股决定脑袋的攻击,我只看到了嫉妒和眼红。一、去医院不看病你去逛街吗&nbsp;去医院你不去看病你去逛街吗?去加油站不加油你去抽烟吗?去部队你不训练战斗技能你去养老吗?来牛客你不努力求职你来干什么来了。&nbsp;牛客本身就是个求职平台,大家分享有用的知识,分享面经,分享offer,分享求职经验的,来牛客不就干这个来了吗?有什么问题吗?...
给个好点的工作吧啊啊...:不知道我看的是不是和博主同样的帖子,我记得原帖是表达的是有些匿名老是发几十万的offer侮辱价,然后就有牛友觉得凡尔赛了导致后面的评论有些偏激。我觉得这个最近闫学晶那个事情有点类似了,她说他儿子一年只能赚七八十万家庭生活都难以为继,不说普通家庭,多少大厂的程序员都赚不到这个数字,大部分家庭看到这种发言肯定会难受的一p,生活的担子又这么重,人都是需要发泄情绪的,互联网就是个极佳的载体,所以很多人直接就喷她了,人在情绪发泄的时候是不思考的,否则就不叫发泄了。然后还有一个点,段哥假定了这些喷的人全都是“躺平的”,这点可能有失偏颇,很多人一直在努力,但是始终缺乏天时地利人和的某一个条件,这点相信段哥找工作的过程中深有体会。绝大部分人都以结果的失败去否认了努力的全过程,可能只是别人努力的方向错了。就像一次面试,可能你准备了很久,结果面试官就是比较奇葩,一直问没有学习到的领域或知识点,然后有人凭一个挂掉的结果就直接给你扣了一个“躺平”的帽子,觉得挂掉是你不够努力,您心里滋味如何?再说点近点的,我也是od,多少同事深夜无偿加班,涨过一分工资吗?多少外包的技术大牛因为学历被困在外包,连od都进不去,这些人难道不努力吗?只是限制与生活、公司制度等等之类的无奈罢了。说到努力,又想到李家琦79元眉笔事件,这么多年有没有认真工作?有没有涨工资?他嘴里说出来是那么的理所当然,打工牛马都知道“任劳任怨”,“认真工作”真能涨工资?只干活不发声就等着被摘果子吧,企业里永远都是“汇报杰出者”升的最快(当然不是所有企业),这种事情相信段哥包括我甚至大部分od都经历过。最近辞职回老家,和老爸散步每次他都会感慨街上的蔬菜小贩多不容易,他们晚上就窝在那种三轮小货车的驾驶室里,腿都伸不直,我们这里晚上零下了,只盖一条薄毛毯,始终舍不得住我们镇上几十块的酒店,因为一车蔬菜就赚几百块顶多一千而且要卖好久,这样的例子还有太多了。这种芸芸众生可能辛苦了一天之后,打开手机看到网上的凡尔赛发言,跟风喷了几句发泄情绪,我觉得这种人不应该扣上“躺平”的帽子。我觉得大部分正常人都是努力的,或者曾经努力过,但世界上有太多努力解决不了的无奈了,甚至说你都没有那个努力的机会,不过正因如此,才显得坚持不懈的努力奋斗之人的难得可贵,认清生活的真相后仍然热爱生活,敢于直面现实的淋漓。
段段STEADY觉醒与突...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务