剑指offer 面试题. 按之字形顺序打印二叉树

题目描述

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

 

方法1：

正常层次遍历，利用普通队列。逢奇数行（从0算起）就把该层结果逆序。

 class Solution {
 public:
     vector<vector<int> > Print(TreeNode* pRoot) {
         if(pRoot==nullptr){return {};}
         queue<TreeNode*> my_queue;
         my_queue.push(pRoot);
         vector<vector<int>>res;
         vector<int> vec;
         int cnt=;
         while(not my_queue.empty()){
             vec.clear();
             int cur_siz=my_queue.size();
             for(int i=;i<cur_siz;++i){
                 auto cur=my_queue.front();
                 my_queue.pop();
                 vec.push_back(cur->val);
                 if(cur->left){
                     my_queue.push(cur->left);
                 }
                 if(cur->right){
                     my_queue.push(cur->right);
                 }
             }
             if(cnt&){
                 reverse(vec.begin(),vec.end());
             }
             res.push_back(vec);
             ++cnt;
         }
         return res;
     }
 };

方法2：

利用双端队列，队列中顺序始终是正常的，但奇数行从右向左遍历，pop右边的，孩子push进左边。偶数行从左向右遍历，pop左边的，孩子push进右边。

 class Solution {
 public:
     vector<vector<int> > Print(TreeNode* pRoot) {
         if(pRoot==nullptr){return {};}
         deque<TreeNode*> my_queue;
         my_queue.push_back(pRoot);
         vector<vector<int>>res;
         int cnt=;
         while(not my_queue.empty()){
             res.push_back({});
             int cur_siz=my_queue.size();
             decltype(pRoot) cur;
             for(int i=;i<cur_siz;++i){
                 if(cnt&){
                     cur=my_queue.back();
                     my_queue.pop_back();
                     if(cur->right){
                         my_queue.push_front(cur->right);
                     }
                     if(cur->left){
                         my_queue.push_front(cur->left);
                     }
                 }
                 else{
                     cur=my_queue.front();
                     my_queue.pop_front();
                     if(cur->left){
                         my_queue.push_back(cur->left);
                     }
                     if(cur->right){
                         my_queue.push_back(cur->right);
                     }
                 }
                 res.back().push_back(cur->val);
             }
             ++cnt;
         }
         return res;
     }
 };