【按之字形顺序打印二叉树】队列数据结构的使用

一、题目

力扣题目链接：link

二、解法

该题需要使用一种数据结构—队列来保存二叉树每一层的元素，队列是只允许在一端进行插入操作，另一端进行删除操作的线性表。允许插入的一端为队尾，允许删除元素的一端为队头。在C++中使用queue<type>来创建队列，常用的函数有：

• q.size(): 队列q的大小
• q.pop(): 删除队列q的队头元素
• q.front(): 队列q的队头元素
• q.push(val): 在队列q的队尾插入元素val
• q.empty(): 队列q是否为空

解题过程如下：
1）创建res保存最终结果，并判断特殊情况：如果pRoot为空，返回空vector；
2）创建队列q以保存二叉树的每一层的节点，创建flag表示从左到右打印还是从右到左打印；
3）队列q的初始元素为pRoot，当队列q不为空时，我们遍历q的每一个元素并进行如下操作：

• 提取队列q的队头元素temp，并删除队头元素，如果temp的左右孩子不为空时，在队列q的队尾插入这两个节点，我们做的操作就是删除队头的节点，同时在队尾添加它的两个左右孩子
• 当flag%2为不同值时，将其以不同的方式添加到数组arr中arr.push_back(temp->val)为从左到右添加，arr.insert(arr.begin(), temp->val)是指在arr的初始位置插入temp->val，即从右到左添加
• 当遍历结束后，队列q中二叉树上一层的节点都被删除并被添加到arr中，下一层的节点都被插入队尾

4）把arr添加到数组res中

/*
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>> res;
        if (pRoot == nullptr) return {};
        queue<TreeNode*> q;
        int flag = 0;
        q.push(pRoot);
        while (!q.empty()){
            vector<int> arr;
            int size = q.size();
            for (int i=0; i < size; i++){
                TreeNode* temp = q.front();
                q.pop();
                if (temp == nullptr) continue;
                if (temp->left) q.push(temp->left);
                if (temp->right) q.push(temp->right);
                if (flag % 2 == 0){
                    arr.push_back(temp->val);
                }else{
                    arr.insert(arr.begin(), temp->val);
                }
            }
            flag++;
            res.push_back(arr);
        }
        return res;
    }
    
};