面试题32：之字形打印二叉树

题目描述

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

考察点：二叉树和栈的理解；新概念的快速理解。

思路：奇数层顺序输出，偶数层逆序输出。想到了两个栈来存放数据。书上写的有点不太明白，自己写了一个两个栈交替的程序。

奇数层从右节点开始入栈2，偶数层从左孩子开始入栈1。

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>> result;
        if (pRoot == NULL)
            return result;

        stack<TreeNode*> stack1;
        stack<TreeNode*> stack2;//初始化两个栈

        stack1.push(pRoot);//根节点入第一个栈
        int flag = 1;//初始化为奇数层
        while (!stack1.empty() || !stack2.empty()) //不为空就是有数据，一直循环输出
        {
            //两个临时vector，存放一行。
            vector<int> re1;//奇数行
            vector<int> re2;//偶数行

            if (flag == 1)
            {
                while (!stack1.empty())
                {
                    TreeNode* pNode = stack1.top();
                    stack1.pop();
                    re1.push_back(pNode->val);//元素出栈入数组

                    if (pNode->left != NULL) //孩子节点入栈2，先左后右
                        stack2.push(pNode->left);

                    if (pNode->right != NULL)
                        stack2.push(pNode->right);
                }
                result.push_back(re1);//结果入二维数组
                flag = 0; //修改层的标志为偶
            }
            if (flag == 0)//操作同栈1
            {
                while (!stack2.empty())
                {
                    TreeNode* pNode = stack2.top();
                    stack2.pop();
                    re2.push_back(pNode->val);

                    if (pNode->right != NULL)//孩子节点入栈1，先右后左
                        stack1.push(pNode->right);

                    if (pNode->left != NULL)
                        stack1.push(pNode->left);    
                }
                if(!re2.empty())//这里需要判断是不到了最后一层。否则会出现讲空元素加入二维数组
                    result.push_back(re2);
                flag = 1;//修改层的标志为奇
            }
        }
        return result;
    }

};