剑指 Offer 32 - III. 从上到下打印二叉树 III
难度中等
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
提示:
1. 节点总数 <= 1000
思路1.0:
做过的:直接来!
代码1.0:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if (root == NULL) return vector<vector<int>>();
vector<vector<int>> rst;
queue<TreeNode*> que;
int layers = 0; //记录层数奇偶
que.push(root);
while (!que.empty())
{
vector<int> thisLayer; //存储该层结果
++layers; //层数+1
int len = que.size(); //处理该层所有节点
for (int i = 0; i < len; ++i)
{
if (que.front()->left != NULL) que.push(que.front()->left);
if (que.front()->right != NULL) que.push(que.front()->right);
//如果是偶数层,从右往左插入
if (layers % 2 == 0)
{
thisLayer.insert(thisLayer.begin(), que.front()->val);
}
//如果是奇数层,从左往右插入
else if(layers%2!=0)
{
thisLayer.push_back(que.front()->val);
}
//该节点处理完毕
que.pop();
}
//该层处理完毕
rst.push_back(thisLayer);
}
return rst;
}
};