按之字形顺序打印二叉树

题目描述

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

/*

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> > vec;

         if(pRoot==NULL) return vec;

        stack<TreeNode *> stack1,stack2;

        stack1.push(pRoot);

        while(!stack1.empty()||!stack2.empty())

            {

              vector<int> data;

               TreeNode* pNode=NULL;

               if(!stack1.empty())

                 {

                 while(!stack1.empty())

                     {

                     pNode=stack1.top(); 

                     data.push_back(pNode->val);

                     stack1.pop(); 

                     if(pNode->left!=NULL)

                     stack2.push(pNode->left);

                     if(pNode->right!=NULL)

                     stack2.push(pNode->right);               

                     }

                

                 }

             else if(!stack2.empty())

                {

                 while(!stack2.empty())

                    {

                     

                     pNode=stack2.top(); 

                     data.push_back(pNode->val);

                     stack2.pop();     

                     

                     if(pNode->right!=NULL)

                     stack1.push(pNode->right);

                     if(pNode->left!=NULL)

                     stack1.push(pNode->left);                      

                    }

                     

               

                 

               }

            

               vec.push_back(data);

            }

        return vec;

           

    }

    

};