剑指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=0;
        while(not my_queue.empty()){
            vec.clear();
            int cur_siz=my_queue.size();
            for(int i=0;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&1){
                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=0;
        while(not my_queue.empty()){
            res.push_back({});
            int cur_siz=my_queue.size();
            decltype(pRoot) cur;
            for(int i=0;i<cur_siz;++i){
                if(cnt&1){
                    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;
    }
};