把二叉树打印成多行
题目描述:
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行
c++中empty的用法:
http://www.cplusplus.com/reference/vector/vector/empty/
思路都在代码里面,这个和把二叉树从上到下打印是一回事,只是需要一个分层的条件
/*
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> > vec;//容器
if(pRoot == NULL) return vec;
queue<TreeNode*> q;//队列
q.push(pRoot);//入队
while(q.empty()==0)
{
int lo = 0, hi = q.size();//队列中元素个数
vector<int> c;//容器,存储数据
while(lo++ < hi)//按层输出打印
{
TreeNode *t = q.front();//取队顶元素
q.pop();//出队
c.push_back(t->val);
//如果有左孩子,右孩子,就让左右孩子入队
if(t->left!=NULL)
q.push(t->left);
if(t->right!=NULL)
q.push(t->right);
}
vec.push_back(c);//将按层打印的数据装入vec容器中
}
return vec;
}
};