按之字形顺序打印二叉树

题目描述

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

 

思路

层序遍历二叉树只需要一个队列即可。而按照题意,奇数从左到右,偶数层从右到左,则利用一个队列和一个栈来实现。队列先进先出,栈先进后出,所以队列保存奇数层结点,栈保存偶数层结点。由于偶数层是从右到左,所以栈直接从左到右即可。而奇数层需要从右到左,所以将奇数层结点按照从右到左插入到队列的头部。

 

代码

/*
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>> res;
        if (pRoot == nullptr) return res;
        bool flag = true;
        stack<TreeNode*> s;
        deque<TreeNode*> q;
        q.push_back(pRoot);
        while (!q.empty() || !s.empty()){
            vector<int> tmp;
            if (flag){
                int size = q.size();
                for (int i = 0; i < size; ++i){
                    auto cur = q.front();
                    tmp.emplace_back(cur->val);
                    q.pop_front();
                    if (cur->left) s.emplace(cur->left);
                    if (cur->right) s.emplace(cur->right);
                }
            }
            else{
                int size = s.size();
                for (int i = 0; i < size; ++i){
                    auto cur = s.top();
                    tmp.emplace_back(cur->val);
                    s.pop();
                    if (cur->right) q.push_front(cur->right);
                    if (cur->left) q.push_front(cur->left);
                }
            }
            res.emplace_back(tmp);
            flag = !flag;
        }
        return res;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值