二叉树的层序遍历比较基础,想法很简单,就是用一个queue就可以实现,一次把弹出节点的左右子节点放入queue中,然后读出就行。
1.简单的二叉树层序遍历
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> res;
if(root==NULL)
return res;
queue<TreeNode*> que;
que.push(root);
while(!que.empty())
{
TreeNode* t=que.front();
que.pop();
res.push_back(t->val);
if(r->left!=nullptr) que.push(r->left);
if(t->right!=nullptr) que.push(r->right);
}
return res;
}
};
2.二叉树层序遍历并逐层打印(每一层输出一行)
注意,这个是要每层输出一行,难点就是控制行的打印。
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int> > res;
if(pRoot==NULL)
return res;
queue<TreeNode*> que;
que.push(pRoot);//先将头结点放入
while(!que.empty())
{
int s=0;
int node_size=que.size();//这一步比着上一步是最重要的区别,目的是
//先读出queue里面还有多少个节点,这个节点数就是每层的数,下面每层打印的时候
//会把打印完的一层逐个扔掉。
vector<int> r;
while(s++<node_size)//s从0开始直到将queue中的数遍历完。
{
TreeNode* t=que.front();
que.pop();
r.push_back(t->val);
if(t->left) que.push(t->left);
if(t->right) que.push(t->right);
}
res.push_back(r);
}
return res;
}
};
3.之字型打印二叉树
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推
解:和前面思路一样,不同的是用两个栈就行了,在这里其实不用计数,直接是a弹出完结束,b弹出完结束。
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int> > res;
if(pRoot==NULL)
return res;
stack<TreeNode*> a;
stack<TreeNode*> b;
a.push(pRoot);
while(!a.empty()||!b.empty())
{
int count1=0;
int nodeInStack1=a.size();
if(nodeInStack1==0)
break;
vector<int> r1;
while(count1++<nodeInStack1)//不用计数,直接while(!a.empty())就可以了
{
TreeNode* t=a.top();
a.pop();
r1.push_back(t->val);
if(t->left) b.push(t->left);
if(t->right) b.push(t->right);
}
res.push_back(r1);
int count2=0;
int nodeInStack2=b.size();
if(nodeInStack2==0)
break;
vector<int> r2;
while(count2++<nodeInStack2)
{
TreeNode* t=b.top();
b.pop();
r2.push_back(t->val);
if(t->right) a.push(t->right);
if(t->left) a.push(t->left);
}
res.push_back(r2);
}
return res;
}
};