103. 二叉树的锯齿形层序遍历

最新推荐文章于 2021-11-23 16:04:38 发布

原创最新推荐文章于 2021-11-23 16:04:38 发布 · 124 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种特殊的二叉树层次遍历算法：先按层次遍历二叉树，再根据层次的奇偶性决定是否反转该层节点值的顺序。通过这种方式，可以实现一种类似于锯齿形的层次遍历效果。

层次优先遍历 BFS

先进行层次遍历，然后区分奇数偶数层（偶数层倒序输出）
在这里插入图片描述

class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> result;
        queue<TreeNode*> tmpQ;
        int isRev = 0;

        if(root == nullptr) 
            return result;
        
        tmpQ.push(root);
        while(tmpQ.size()!=0)
        {
            int len = tmpQ.size();
            vector<int>levelResult;
            for(int i=0;i<len;i++)
            {
                TreeNode* cur=tmpQ.front();
                tmpQ.pop();
                levelResult.push_back(cur->val);
                if(cur->left!=nullptr) tmpQ.push(cur->left);
                if(cur->right!=nullptr) tmpQ.push(cur->right);
            }
            if(isRev == 0) 
            {
                result.push_back(levelResult);
                isRev = 1;
            }
            else
            {
                reverse(levelResult.begin(),levelResult.end());//反转当前层序列
                result.push_back(levelResult);
                isRev = 0;
            }
        }
        return result;
    }
};