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

题目描述

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

思路：

利用两个栈 stack1和stack2，分别存放奇数行结点和偶数行结点；设定行号变量，例如layer来表示奇数行或者偶数行。

当打印stack1中的结点的同时，将该结点的子树按照先左子树后右子树的顺序压入stack2；

当打印stack2中的结点的同时，将该结点的子树按照先右子树后左子树的顺序压入stack1。

自己画个图就知道了。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    vector > Print(TreeNode* pRoot) {
        vector > res;
        if(!pRoot)
            return res;
        stack stack1;
        stack stack2;
        int layer = 1;
        stack1.push(pRoot);
        while(!stack1.empty() || !stack2.empty())
        {
            vector tmp;
            TreeNode* node;
            if(layer&0x1)
            {
                while(!stack1.empty())
                {
                    node = stack1.top();
                    stack1.pop();
                    tmp.push_back(node->val);
                    if(node->left)
                        stack2.push(node->left);
                    if(node->right)
                        stack2.push(node->right);
                }
            }
            else
            {
                while(!stack2.empty())
                {
                    node = stack2.top();
                    stack2.pop();
                    tmp.push_back(node->val);
                    if(node->right)
                        stack1.push(node->right);
                    if(node->left)
                        stack1.push(node->left);
                }
            }
            res.push_back(tmp);
            ++layer;
        }
        return res;
    }
};