题目:
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路:
同样地,使用队列依然可以解决问题,而且,我们还需要一个变量来指示是否已经打印完一层。
程序代码:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int> > r;
vector<int> rr; //存储单行需要压入的值
if(!pRoot)
return r;
queue<TreeNode*> q;
q.push(pRoot);
int num_next=0; //表征下一层的结点数
int num_now=1; //表征当前行还没有打印的结点数
//int row=0;
while(q.size())
{
TreeNode *node=q.front();
rr.push_back(node->val);
cout<<node->val<<" ";
if(node->left)
{
q.push(node->left);
++num_next;
}
if(node->right)
{
q.push(node->right);
++num_next;
}
q.pop();
--num_now;
if(num_now==0)
{
r.push_back(rr);
vector <int>().swap(rr); //清空rr的内容
cout<<endl;
//++row;
num_now=num_next;
num_next=0;
}
}
return r;
}
运行结果: