面试题32-2:把二叉树打印多行

从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
class Solution {
public:
    /*
    思路:改编于面试题32,利用队列。
    先打印根节点,并把根节点左右子节点入队,打印队首元素并出队,将其左右孩子入队...循环如此一直到队列为空。
    此题要分层打印,则添加一个一维vector数组levelPrint存储每层的节点,再增设一个指针rightSon赋值为队列qTree尾元素,指向每层的最右节点
    当当前指针tNode与rightSon指向同一节点,表示该层遍历完毕,打印该层并重新设置levelPrint和rightSon节点
    */
        vector<vector<int> > Print(TreeNode* root) {
            //将vector作为返回值类型
            vector<vector<int> > printTree;
            //若为空树,直接返回空的数组
            if(root==nullptr)
                return printTree;
            //因为要分层打印,所以定义辅助指针指向每层最右边的节点,表示这是这一层的结尾
            TreeNode *rightSon=root;
            //储存每层的节点,最后加入到二维vector printTree中
            vector<int> levelPrint;
            //建立辅助队列
            queue<TreeNode*> qTree;
            TreeNode *tNode=root;
            //根节点入队
            qTree.push(tNode);
            //若队列不为空且树指针不指向空,执行循环:
            while(!qTree.empty()&&tNode!=nullptr)
            {
                //将队列首元素放入打印数组中
                levelPrint.push_back(tNode->val);

                //若队手元素左右子树不为空,压入队列
                if(tNode->left)
                    qTree.push(tNode->left);
                if(tNode->right)
                    qTree.push(tNode->right);
                //若此节点等于当前层最右节点,则将这一层结果输出并将最右节点重新赋值为队列尾结点
                if(rightSon==tNode)
                {
                    printTree.push_back(levelPrint);
                    levelPrint.clear();
                    rightSon=qTree.back();
                }
                //队首元素出队
                qTree.pop();
                //tNode赋值为新的队首元素
                tNode=qTree.front();
            }
            return printTree;
        }    
};
全部评论

相关推荐

12-18 18:50
已编辑
门头沟学院 golang
牛客33637108...:重点是要事已密成,在没有进入这家公司之前,不要有任何的泄露信息,我之前跟你一样,面了一家光伏设备厂,底薪7500加上出差补贴大概有13,000左右,已经给了口头offer了,甚至要了我的在校成绩的所有信息,还向我要了三方的网签二维码,到后面还是毁约了,我干过最愚蠢的事情就是向同学透露要签三方的事,之后的失败只会让他们幸灾乐祸,这是即将结束的大学生活给我的最后一课,不要相信任何的口头三方,该面的就去面,甚至签了三方也有毁约的可能,就像我现在签了三方还在外面实习呢,春招还是要继续参加的,不能停止面试,不然到后面毁三方的时候,重新捡起的面试很麻烦的,这是我一点点小小的见解。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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