leetcode Binary Tree Zigzag Level Order Traversal

给定一个树，按照Z字形记录每一行。例如：

Given binary tree {3,9,20,#,#,15,7},

    3
   / 
  9  20
    /  
   15   7

return its zigzag level order traversal as:

[
  [3],
  [20,9],
  [15,7]
]

这题想了有一会，按照上一题思路的启发。这题也是应该要用某个数据结构会比较好解答。然后就想用队列，然后可以任意从前面删除就好了，可以队列之提供后面pop的操作。于是就想到用栈。

思路：用两个栈，每个栈存一层，例如上一题，第一个栈sta1，先push(3)，然后去sta1的元素左边右边先后入sta2，那么再pop就是20,9了，当然pop的时候又要对其进行push栈sta1，先把7push然后15push，最后栈sta1 pop的时候就是15,7了。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
vector<vector<int> > zigzagLevelOrder(TreeNode *root)
{
    vector<vector<int> > ans;
    if (!root) return ans;

    vector<int> tmp;
    stack<TreeNode *> sta1, sta2;
    TreeNode *p;

    sta1.push(root);
    while(!sta1.empty() || !sta2.empty())
    {
        while (!sta1.empty())
        {
            p = sta1.top();
            sta1.pop();
            tmp.push_back(p -> val);
            if (p -> left)
                sta2.push(p -> left);
            if (p -> right)
                sta2.push(p -> right);
        }
        ans.push_back(tmp);
        tmp.clear();
        if (sta1.empty() && sta2.empty()) return ans;

        while(!sta2.empty())
        {
            p = sta2.top();
            sta2.pop();
            tmp.push_back(p -> val);
            if (p -> right)
                sta1.push(p -> right);
            if (p -> left)
                sta1.push(p -> left);
        }
        ans.push_back(tmp);
        tmp.clear();
    }
    return ans;
}
};