LeetCode: Binary Tree Level Order Traversal

AIvin24

于 2014-07-01 10:28:36 发布

阅读量565

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/AIvin24/article/details/36177981

LeetCode 专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种使用双队列实现的二叉树层次遍历算法，通过交替使用两个队列来保存不同层级的节点，实现了对二叉树的层次遍历并返回每一层的节点值。

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

思路：使用两个队列，一个队列存储当前一层的所有节点，另一个队列存储下一层的节点，结合层次遍历树结构。

code:

class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
        vector<vector<int> > ret;
        if(root!=NULL){
            TreeNode * temp = root;
            queue<TreeNode *> firstQ,nextQ;
            firstQ.push(temp);
            while(!firstQ.empty()){
                vector<int> layerRet;
                while(!firstQ.empty()){
                    TreeNode *t = firstQ.front();
                    if(t->left != NULL)
                        nextQ.push(t->left);
                    if(t->right != NULL)
                        nextQ.push(t->right);
                        
                    layerRet.push_back(t->val);
                    firstQ.pop();
                }
                firstQ = nextQ;
                while(!nextQ.empty())  
                    nextQ.pop();
                ret.push_back(layerRet);
            }
        }
        return ret;
    }
};