二叉树打印多行（队列法）——剑指Offer

题目描述

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

题目分析

使用队列存储每一行打印的结点信息，但是要注意每一行最后的标志判定，我使用的是在队列最后添加NULL进行标志的判定，判定这一行的结点打印结束。

代码

C++代码如下：

/*
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> > res;
            if(!pRoot) return res;
            queue<TreeNode*> q;
            q.push(pRoot);
            q.push(NULL);//结尾标志结束符
            vector<int> tmp;
            while(q.size())
            {
                auto t=q.front();
                q.pop();
                //如果t是NULL
                if(!t)
                {
                    if(tmp.empty())
                        break;
                    res.push_back(tmp);
                    tmp.clear();
                    q.push(NULL);//结尾标志结束符
                    continue;
                }
                tmp.push_back(t->val);
                if(t->left) q.push(t->left);
                if(t->right) q.push(t->right);
            }
            return res;
        }
    
};