【剑指offer-解题系列(61)】把二叉树打印成多行

来源:互联网 发布:淘宝上卖内衣的货源 编辑:程序博客网 时间:2024/05/17 20:00

题目描述

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

分析

层次打印使用队列就行了,和上一篇的方法几乎一模一样,只是堆栈换成队列,另外不需要符号了

代码实现

  vector<vector<int> > Print(TreeNode* root) {

        vector<vector<int> >res;
        if(!root)
            return res;
        TreeNode*p=NULL;
        Q1.push(root);
        while(!Q1.empty() || !Q2.empty()){
            vector<int>r1;
            while(!Q1.empty()){ 
                p = Q1.front();
                r1.push_back(p->val);
                Q1.pop();
                if(p->left ) Q2.push(p->left ); 
                if(p->right) Q2.push(p->right); 
            }
            if(!r1.empty())res.push_back(r1);   
            vector<int>r2;
            while(!Q2.empty()){
                vector<int>r;
                p = Q2.front();
                r2.push_back(p->val);
                Q2.pop();
                if(p->left  ) Q1.push(p->left ); 
                if(p->right ) Q1.push(p->right); 
            }
            if(!r2.empty())res.push_back(r2);
        }
        return res;
    }
    queue<TreeNode*>Q1;
    queue<TreeNode*>Q2;