第三周：[Leetcode]103. Binary Tree Zigzag Level Order Traversal

最新推荐文章于 2024-01-19 11:15:23 发布

line02

最新推荐文章于 2024-01-19 11:15:23 发布

阅读量174

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/line02/article/details/61451012

算法专栏收录该内容

45 篇文章

订阅专栏

本文介绍了一种算法，用于实现二叉树节点值的锯齿形层序遍历。即先从左至右遍历一层节点，再从右至左遍历下一层节点，并交替进行。通过广度优先搜索并利用队列实现，同时借助布尔变量来控制输出顺序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Binary Tree Zigzag Level Order Traversal
Given a binary tree, return the zigzag level order traversal of its nodes’ values. (ie, from left to right, then right to left for the next level and alternate between).
For example:
Given binary tree [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return its zigzag level order traversal as:
[
[3],
[20,9],
[15,7]
]

对二叉树进行广度优先搜索，设置一个布尔变量记录当前层的数据的输出顺序是从左到右还是从右到左。

class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
         vector<vector<int>> ans;
        vector<int> level;
        if(root == NULL)
            return ans;
        queue<TreeNode*> q;
        q.push(root);
        int count = 1,next_count = 0;
        bool left_right = true;
        while(!q.empty()){
            TreeNode* node = q.front();
            if(left_right)
                level.push_back(node -> val);
            else
                level.insert(level.begin(), node -> val);
            if(node -> left != NULL){
                q.push(node -> left);
                next_count ++;
            }
            if(node -> right != NULL){
                q.push(node -> right);
                next_count ++;
            }
            q.pop();
            count --;
            if(count == 0){
                ans.push_back(level);
                level.clear();
                count = next_count;
                next_count = 0;
                left_right = left_right ? false : true;
            }
        }
        return ans;
    }
};