题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
这道题与之前做的几道题的方法一致,借助队列数据结构,采用广度优先遍历的方法,将每一层的兄弟结点依次入队,再依次出出队列,这里需要一个标记记录每一层结点的数目,根据标记的值可以知道一层结点是否打印完毕;细节见代码:
/*
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;
if (pRoot == NULL)
{
return result;
}
queue<TreeNode*> q;
q.push(pRoot);
int len = 0;
while (!q.empty())
{
vector<int> vec;
len = q.size();
while (len--)
{
TreeNode* node = q.front();
vec.push_back(node->val);
q.pop();
if (node->left)
{
q.push(node->left);
}
if (node->right)
{
q.push(node->right);
}
}
result.push_back(vec);
}
return result;
}
};