剑指offer——33分行从上到下打印二叉树

题目描述

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

 

题解：

　　使用BFS，按层打印即可

　　

 class Solution {
 public:
     vector<vector<int> > Print(TreeNode* pRoot) {
         vector<vector<int>>res;
         BFS(pRoot, res);
         return res;
     }
     void BFS(TreeNode *root, vector<vector<int>>&res)
     {
         if (root == nullptr)return;
         queue<TreeNode*>q;
         q.push(root);
         while (!q.empty())
         {
             queue<TreeNode*>temp;
             vector<int>v;
             while (!q.empty())
             {
                 TreeNode* p = q.front();
                 q.pop();
                 v.push_back(p->val);
                 if (p->left != nullptr)temp.push(p->left);
                 if (p->right != nullptr)temp.push(p->right);
             }
             res.push_back(v);
             q = temp;
         }
     }
 };