【剑指offer】把二叉树打印成多行，C++实现

原创文章，转载请注明出处！

本题牛客网地址

博客文章索引地址

博客文章中代码的github地址

1.题目

      从上到下按层打印二叉树，同一层结点从左至右输出，每一层输出一行。例如：下面二叉树的打印结果为：

8

6 10

5 7 9 11

2.思路

      本题目使用队列和两个变量作为辅助，利用队列的先进先出策略，按层遍历二叉树。队列保存将要打印的节点，一个变量表示当前层中还没有打印的节点数，一个节点表示下一层节点的数目。

3.代码

/*
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>>  result; // 存储全部节点
            vector<int> temp;            // 存储某一层节点

            // 边界条件
            if(pRoot == nullptr)
                return result;

            // 辅助容器：队列
            queue<TreeNode*> q;
            TreeNode* fr;
            int now_level=1;
            int next_level=0;

            // 根节点入队列
            q.push(pRoot);

            // 遍历队列
            while(!q.empty())
            {
                // 节点弹出队列
                fr=q.front();
                temp.push_back(fr->val);
                q.pop();

                // 遍历节点左右子树
                if(fr->left != NULL){
                    q.push(fr->left);
                    ++next_level;
                }
                if(fr->right != NULL){
                    q.push(fr->right);
                    ++next_level;
                }
                --now_level;

                // 判断当前层是否打印完
                if(now_level==0){
                    now_level=next_level;
                    next_level=0;
                    result.push_back(temp);
                    temp.clear(); // 清除一维vector
                }
            }

           return result;
        }
};