59.按之字形顺序打印二叉树

题目

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

思路

• 之字形打印二叉树，其实就是按层访问二叉树的一个变种。
• 用双向链表来保存二叉树的结点。
• 第一层就是根节点，别根节点的左、右孩子尾插到队列中。
• 第二层从右向左打印，双向队列从尾部访问，打印各结点，同时各结点右、左孩子头插入双向队列。
• 第三层从左向右打印，双向队列从首部访问，打印各结点，同时各结点左、右孩子尾差入双向队列。
• 停止条件为双向队列为空。
  

代码

/*
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 res;
        
        deque<TreeNode*> dq;
        dq.push_back( pRoot );
        bool leftToRight = true;
        
        while ( !dq.empty() ) {
            int size = dq.size();
            vector<int> temp;
            
            if ( leftToRight ) {
                while ( size-- ) {
                    TreeNode* node = dq.front();
                    dq.pop_front();
                    temp.push_back( node->val );

                    if ( node->left) 
                        dq.push_back( node->left );
                    
                    if ( node->right) 
                        dq.push_back( node->right );
                }
            }
            else {
                while ( size-- ) {
                    TreeNode* node = dq.back();
                    dq.pop_back();
                    temp.push_back( node->val );
                    
                    if ( node->right ) 
                        dq.push_front( node->right );
                    
                    if ( node->left ) 
                        dq.push_front( node->left );
                }
            }
            
            res.push_back( temp );
            leftToRight = !leftToRight;
        }
        
        return res;
    }
    
};