版权声明:喜欢可以转载哦,声明一下哈 https://blog.csdn.net/QQ1910084514/article/details/82704904
题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路
在层序遍历的基础上,设置两个遍历纪录当前层有几个结点,和下一层有几个结点,根据两个变量即可确定是否来到下一层
代码
/*
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> > vecs;
vector<int> vec;
queue<TreeNode*> q;
int count1=1;//当前层的个数
int count2=0;//下一层的个数
if(pRoot==NULL)
return vecs;
q.push(pRoot);
while(q.size()>0)
{
TreeNode* front=q.front();
vec.push_back(front->val);
count1--;
if(front->left!=NULL)
{
q.push(front->left);
count2++;
}
if(front->right!=NULL)
{
q.push(front->right);
count2++;
}
if(count1==0)//标志进入下一层
{
count1=count2;
count2=0;
vecs.push_back(vec);
vec.clear();//清除
}
q.pop();
}
return vecs;
}
};