剑指 Offer 32 - III. 从上到下打印二叉树 III(C++) 层序遍历 + 双端队列(奇偶层逻辑分离)前取后放,后取前放

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

例如:

给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果:

[
  [3],
  [20,9],
  [15,7]
]

提示:

节点总数 <= 1000

解题思路

奇偶层的打印顺序不一样是相反的,可以利用层数偶数与否调用reverse来解决,但是海量数据时这个效率很低,不推荐。

因为奇数层的打印顺序是从左到右,偶数层的打印顺序是从右到左,可以利用STL容器deque中
push_back(),push_front(),front(),back(),pop(),popfront()来实现

前取后放,后取前放

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if (root==NULL)//根节点为空时
            return res;
        bool flag = true; //从左向右打印为true,从右向左打印为false
        deque<TreeNode*> q;//存取元素的队列
        q.push_back(root);//根节点入队列
        while (!q.empty())//队列非空时执行
        {
            int n = q.size();//队列长度
            vector<int> out;//暂存一维数组
            TreeNode* node;//二叉树
            while (n>0)
            {
            //奇数层要求从左向右输出,偶数层反之
            //step 1.一开始flag==ture时,从前边取奇数层(即是根节点),
            //然后存下一层(偶数层)的节点顺序地放在队列尾部(此层节点顺序排列);
            //step 2.到偶数层从尾部取出(存放在数组中就是逆序了),
            //然后存下一层(奇数层)的节点逆序地放在队列头部(此层节点顺序排列);
            //因此,按照此逻辑一层层遍历可以得到结果

                if (flag) // 前取后放:从左向右打印,所以从前边取,后边放入
                // 从队头取出要打印的,左右子节点正序从队尾加入
                {
                    node = q.front();
                    q.pop_front();
                    if (node->left)
                        q.push_back(node->left);  // 下一层顺序存放至尾
                    if (node->right)
                        q.push_back(node->right);
                }
                else  //后取前放: 从右向左,从后边取,前边放入
                // 从队尾取出要打印的,左右子节点逆序从队头加入
                {
                    node = q.back();
                    q.pop_back();
                    if (node->right)
                        q.push_front(node->right);  // 下一层逆序存放至首
                    if (node->left)
                        q.push_front(node->left);
                    //存了奇数层的节点在前面,按照从左向右的顺序方式
                }
                out.push_back(node->val);
                n--;//n自减
            }
            flag = !flag; // 改变打印顺序
            res.push_back(out);// 打印完了一层
        }
        return res;
    }
};

复杂度分析:

时间复杂度 O(N):
空间复杂度 O(N) :
作者:bryceustc
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/solution/cli-yong-dequeshuang-duan-dui-lie-hao-shi-0ms-ji-b/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值