剑指offer-59. 按之字形顺序打印二叉树

题目描述

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

解题思路：

 1、准备两个栈存放结点
2、打印奇数层时，先保存左子结点，再保存右子结点到第一个栈中
3、打印偶数层时，先保存右子结点，再保存左子结点到第二个栈中
4、打印完一个栈里的结点时，交换这两个栈，继续打印下一层。

代码实现：

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
#include<stack>
class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        /*
        解题思路：
        1、准备两个栈存放结点
        2、打印奇数层时，先保存左子结点，再保存右子结点到第一个栈中
        3、打印偶数层时，先保存右子结点，再保存左子结点到第二个栈中
        4、打印完一个栈里的结点时，交换这两个栈，继续打印下一层。
        */
        vector<int> result;
        vector<vector<int> > ans;
        if(pRoot == NULL){
            return ans;
        }
        stack<TreeNode*> levels[2];
        int current = 0,next = 1;
        levels[current].push(pRoot);
        while(!levels[0].empty() || !levels[1].empty()){
            TreeNode* currentNode = levels[current].top();
            levels[current].pop();
            result.push_back(currentNode->val);
            
            if(current == 0){//打印奇数层时，先保存左子结点，再保存右子结点到第一个栈中
                if(currentNode->left!=NULL){
                    levels[next].push(currentNode->left);
                }
                if(currentNode->right!=NULL){
                   levels[next].push(currentNode->right); 
                }
            }else{//打印偶数层时，先保存右子结点，再保存左子结点到第二个栈中
               if(currentNode->right!=NULL){
                   levels[next].push(currentNode->right); 
                } 
                if(currentNode->left!=NULL){
                    levels[next].push(currentNode->left);
                }
            }
            if(levels[current].empty()){//打印完一个栈里的结点时，交换这两个栈，继续打印下一层。
                ans.push_back(result);
                result.clear();
                current = 1 - current;
                next = 1 - next;
            }
        }
        return ans;
        
    }
    
};

 效率：