按之字形顺序打印二叉树/把二叉树打印成多行

按之字形顺序打印二叉树/把二叉树打印成多行

题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

解题思路

• 可以用一个队列和一个flag来表示
• flag用来表示是奇数层还是偶数层（根节点是第0层）
• 如果是偶数层，vector翻转插入即可
• 当前层的size记录好就可以正确表示
• 这两道题都是一种思路
• 除了如果是之行打印的话 奇数行翻转一下，然后追加；普通打印，直接追加即可

/*
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 == NULL){
            return res;
        }
        queue<TreeNode*> q;
        q.push(pRoot);
        bool isEven = false;
        while(!q.empty()){
            int size = q.size();     //这一层的size 
            vector<int> temp;
            for(int i = 0; i < size; i++){
                temp.push_back(q.front() -> val);
                TreeNode* cur = q.front();
                q.pop();
                if(cur -> left != NULL) 
                    q.push(cur -> left);             
                if(cur -> right != NULL){ 
                    q.push(cur -> right);            
                }
            }
            if(isEven){
                std::reverse(temp.begin(), temp.end()); 	// 普通打印的话，就不用翻转了，直接push_bakc即可
            }
            isEven = !isEven;
            res.push_back(temp);           
        }
        return res;
    }
    
};