【102.Binary Tree Level Order Traversal】
「思路」
运用广度优先算法(BFS)的思想。用一个队列q,在处理并pop上一层的数据的同时push近其下一层左右子树,pop完上一层时,下一层的数据也刚好完全push入队列,以此类推完成操作。
「运行结果」
「代码」
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>>ans;
if(!root)return ans;
queue <TreeNode*> q;
q.push(root);
while(!q.empty())
{
int num=q.size();
vector<int> group;
for(int i=0;i<num;i++)
{
TreeNode*t=q.front();
group.push_back(t->val);
q.pop();
if(t->left) q.push(t->left);
if(t->right)q.push(t->right);
}
ans.push_back(group);
}
return ans;
}
};
【637.Average of Levels in Binary Tree】
「思路」
同样是典型的BFS题,与上题思路大致相同,就是多加了一个总和在循环里。
「运行结果」
「代码」
class Solution {
public:
vector<double> averageOfLevels(TreeNode* root) {
vector<double>ans;
queue<TreeNode*>q;
q.push(root);
while(!q.empty())
{
int num=q.size();
double sum=0;
for(int i=0;i<num;i++)
{
TreeNode*t=q.front();
q.pop();
if(t->left) q.push(t->left);
if(t->right)q.push(t->right);
sum+=t->val;
}
ans.push_back(sum/num);
}
return ans;
}
};
【103. Binary Tree Zigzag Level Order Traversal】
「思路」
在102的基础上,最后加一个用于反转代码,格一行将数据反转一次。
「运行结果」
「代码」
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>>ans;
if(!root)return ans;
queue <TreeNode*> q;
q.push(root);
//int z=0;
while(!q.empty())
{
int num=q.size();
vector<int> group;
for(int i=0;i<num;i++)
{
TreeNode*t=q.front();
group.push_back(t->val);
q.pop();
if(t->left) q.push(t->left);
if(t->right)q.push(t->right);
}
ans.push_back(group);
}
for(int i=1;i<ans.size();i=i+2)
{
reverse(ans[i].begin(),ans[i].end());
}
return ans;
}
};