剑指 Offer 32 - III. 从上到下打印二叉树 III

#include<iostream>
#include<queue>
using namespace std;

 struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 };
 //在II的基础上加一个奇数层偶数层 反转操作  比较简单
class Solution {
    queue<TreeNode*> q;//利用队列进行广度优先搜索形成层序遍历
    vector<vector<int>> m;//储存值
    vector<int>t;//二维数组vector必须进行整体赋值
    int count1=0;//奇数从左到右 偶数从右到左
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(root==NULL)//如果根节点为空就不用搜了
        return m;
        auto ED=new TreeNode(-1000);//每一层尾节点 判断是否到下一层
        q.push(root);
        q.push(ED);
        while(!q.empty())//队列遍历模板
        {
            auto p=q.front();
            q.pop();
            if(p!=NULL&&p!=ED)//不为空和尾节点时赋值
            {
                t.push_back(p->val);
                q.push(p->left);
                q.push(p->right);
            }
            if(p==ED)
            {
                count1++;
                if(count1%2==0)reverse(t.begin(),t.end());//偶数反转一下
                if(t.size()!=0)//不为空时进行赋值
                m.push_back(t);
                t.clear();//清空
                if(q.size()!=0)//不为空时加个尾节点
                q.push(ED);
            }
        }
        return m;
    }
};