按之字形顺序打印二叉树

题目描述

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

分析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>> vvi;        vector<int> vi;        if(!pRoot) return {};        stack<TreeNode*> p1,p2;   //分别用两个不同的栈存储奇数和偶数层的结点        p1.push(pRoot);        int level=0;      //二叉树的深度记录变量        while(!p1.empty()||!p2.empty())  //直到所有节点都出栈，跳出循环            {              ++level;              if(level%2==1)                 {                 while(!p1.empty())                  {                    int len=p1.size();                    while(len--)                      {                        TreeNode* tem=p1.top();  //用一个变量记录栈顶节点，即每一层中的结点                        p1.pop();                  //删除节点                        vi.push_back(tem->val);   //将每一层节点的int数值存入vi                        //当层数为奇数时，先左孩子后右孩子；                        if(tem->left) p2.push(tem->left);   //将下一层的结点存储在另一个栈里                        if(tem->right) p2.push(tem->right);                          }//如果存储在同一个栈里，当一层中的结点大于一个时，这一层的结点还没有完全取出来时，就会存储下一层的结点，                     //  导致取出的栈顶元素有一部分是下一层的元素                 }              }            else{                while(!p2.empty())                    {                    int len=p2.size();                    while(len--)                        {                        TreeNode* tem=p2.top();  //用一个变量记录栈顶节点，即每一层中的结点                        p2.pop();                  //删除节点                        vi.push_back(tem->val);   //将每一层节点的int数值存入vi                        //当层数为偶数时，先右孩子后左孩子；                        if(tem->right) p1.push(tem->right);                        if(tem->left) p1.push(tem->left);                    }                }            }            vvi.push_back(vi);            vi.clear();        }        return vvi;    }};

分析二：也可以将二叉树按层遍历后，将偶数层的序列颠倒输出

class Solution {public:    vector<vector<int> > Print(TreeNode* pRoot) {        vector<vector<int>> res;        if(pRoot == NULL)            return res;        queue<TreeNode*> que;        que.push(pRoot);        bool even = false;        while(!que.empty()){            vector<int> vec;            const int size = que.size();            for(int i=0; i<size; ++i){                TreeNode* tmp = que.front();                que.pop();                vec.push_back(tmp->val);                if(tmp->left != NULL)                    que.push(tmp->left);                if(tmp->right != NULL)                    que.push(tmp->right);            }            if(even)                std::reverse(vec.begin(), vec.end());            res.push_back(vec);            even = !even;        }        return res;    }     };