剑指offer--60.把二叉树打印成多行

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。
时间限制：1秒 空间限制：32768K 热度指数：111808

思路

见代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
//层次遍历，用一个队列，同时用三个变量记录，当点结点位置，
//当前层结尾结点位置，下一层结尾结点位置
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) 
        {
            vector<vector<int> > ans;
            if(pRoot==NULL)
                return ans;
            queue<TreeNode*> q;
            q.push(pRoot);
            int cnt=1;    //此时访问的结点
            int cc=1;    //当前层
            int next=1;//下一层
            vector<int> t;
            while(!q.empty())
            {
                t.push_back(q.front()->val);
                if(q.front()->left!=NULL)
                {//每一个结点进队，表示下一层结点结尾位置+1
                    q.push(q.front()->left);
                    ++next;
                }
                if(q.front()->right!=NULL)
                {//每一个结点进队，表示下一层结点结尾位置+1
                    q.push(q.front()->right);
                    ++next;
                }
                q.pop();
                if(cnt==cc)
                {//若当前结点已经到当前层的尾端时，将当前层尾端值更新为下一层尾端
                    ans.push_back(t);
                    cc=next;
                    t.clear();
                }
                ++cnt;
            }
            return ans;
        }
    
};