Leetcode-102. Binary Tree Level Order Traversal

最新推荐文章于 2024-06-25 11:36:41 发布

原创最新推荐文章于 2024-06-25 11:36:41 发布 · 307 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #层次遍历 #队列 #leetcode

数据结构及算法同时被 2 个专栏收录

111 篇文章

订阅专栏

树

20 篇文章

订阅专栏

本文介绍了一种解决二叉树层次遍历问题的方法。通过使用两个队列交替存储每层节点的方式，实现了从左到右按层输出二叉树节点值的功能。此算法适用于计算机科学中的数据结构学习。

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

题目

Given a binary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).

For example:
Given binary tree [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return its level order traversal as:
[
[3],
[9,20],
[15,7]
]
输出二叉树层次遍历结果

思路

设置两个队列，将相同层的节点存储在一个队列中，然后再将所有下层的节点放到另一个队列中，循环执行

代码

class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode* root) {
        vector<vector<int> >res;
        if(!root)
            return res;
        queue<TreeNode*> que[2];
        int flag = 0;
        que[flag].push(root);
        while(!que[flag].empty()) {
            vector<int> cur;
            while(!que[flag].empty()) {
                TreeNode* t = que[flag].front();
                que[flag].pop();
                cur.push_back(t->val);
                if(t->left)
                    que[(flag+1)%2].push(t->left);
                if(t->right)
                    que[(flag+1)%2].push(t->right);
            }
            res.push_back(cur);
            flag = (flag+1)%2;
        }
        return res;
    }
};