队列+宽搜(bfs)

目录

一:N叉树的层序遍历

1.1题目

1.2算法原理

1.3代码

二:二叉树的锯齿形层序遍历

2.1题目

2.2算法原理

2.3代码

三:二叉树最大宽度

3.1题目

3.3算法原理

3.3代码

四:在每个树中找最大值

4.1题目

4.2算法原理

4.3代码


一:N叉树的层序遍历

1.1题目

题目链接:https://leetcode.cn/problems/n-ary-tree-level-order-traversal/


1.2算法原理


1.3代码

cpp 复制代码
class Solution {
public:
    vector<vector<int>> levelOrder(Node* root) 
    {
        vector<vector<int>> ret;//记录最终结果
        queue<Node*> q;//层序变量所需要的队列
        
        if(root == nullptr) return ret;

        q.push(root);
        while(q.size())
        {
            int sz = q.size();//统计本层的节点个数
            vector<int> tmp;//统计本层的值

            for(int i = 0; i <sz;i++)
            {
                Node* t = q.front();
                q.pop();
                tmp.push_back(t->val);//把节点的值加入数组
                //把孩子节点加入队列
                for(Node* child : t->children)
                    if(child != nullptr)
                        q.push(child);
            }
            ret.push_back(tmp);
        }

        return ret;
    }
};

二:二叉树的锯齿形层序遍历

2.1题目

题目链接:https://leetcode.cn/problems/binary-tree-zigzag-level-order-traversal/description/


2.2算法原理

锯齿形的层序遍历与第一题的层序遍历唯一不同的是,如果认为头结点是第一层,每一个奇数层都是正常层序遍历的逆序,也就是每次到奇数层的时候,先正常层序遍历,最后push到结果之前逆序即可


2.3代码

cpp 复制代码
class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) 
    {
        vector<vector<int>> ret;
        queue<TreeNode*> queue;
        if(root == nullptr) return ret;
        
        queue.push(root);
        int cont = 0;
        while(queue.size())
        {
            int n = queue.size();//当前层要遍历的次数
            vector<int> tmp;//统计当前层的值

            for(int i = 0; i < n;i++)
            {
                TreeNode* cur = queue.front();
                queue.pop();
                tmp.push_back(cur->val);
                if(cur->left)  queue.push(cur->left);
                if(cur->right) queue.push(cur->right);
            }
            if(cont%2) reverse(tmp.begin(),tmp.end());
            ret.push_back(tmp);
            cont++;
        }   

        return ret;
    }
};

三:二叉树最大宽度

3.1题目

题目链接:https://leetcode.cn/problems/maximum-width-of-binary-tree/


3.3算法原理


3.3代码

cpp 复制代码
class Solution {
public:
    int widthOfBinaryTree(TreeNode* root) 
    {
        vector<pair<TreeNode*,unsigned int>> q;//使用数组模拟栈,方便拿去队尾元素
        unsigned int ret = 0;
        q.push_back({root,1});
        while(q.size())
        {
            int n = q.size();//统计该层元素个数
            auto& [x1,y1] = q[0];//队头
            auto& [x2,y2] = q.back();//队尾
            ret = max(ret,y2-y1+1);

            vector<pair<TreeNode*,unsigned int>> tmp;
            for(auto &[x,y] : q)
            {
                if(x->left) tmp.push_back({x->left,y*2});
                if(x->right) tmp.push_back({x->right,y*2+1});
            }
            q = tmp;//更新数组
        }    
        return ret;
    }
};

四:在每个树中找最大值

4.1题目

题目链接:https://leetcode.cn/problems/find-largest-value-in-each-tree-row/


4.2算法原理

与第一个题目的层序遍历原理相同,只是把每层的值加入数组逻辑改为比较每层值的大小取最大值


4.3代码

cpp 复制代码
class Solution {
public:
    vector<int> largestValues(TreeNode* root) 
    {
        vector<int> ret;
        if(root == nullptr) return ret;
        queue<TreeNode*> q;
        q.push(root);
        while(q.size())
        {
            int n = q.size();
            int val = INT_MIN;
            for(int i = 0; i < n;i++)
            {
                TreeNode* cur = q.front();
                q.pop();
                val = max(val,cur->val);
                if(cur->left) q.push(cur->left);
                if(cur->right) q.push(cur->right);
            }
            ret.push_back(val);
        }
        return ret;
    }
};
相关推荐
倒头就睡的小比特3 天前
算法竞赛C++常用的STL
c++·算法
weilx12343 天前
C++笔记-文件IO-<fcntl.h>
c++
小羊没烦恼!3 天前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
猎头南楼3 天前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
Smileyqp沛沛3 天前
前端?C++ ?较大差异基础罗列
c++·基础·前端转c++
m0_547486663 天前
《数据结构教程》全套 PPT课件2026
数据结构
C语言小火车3 天前
C/C++ 为什么需要编译器?
开发语言·c++
旖旎夜光3 天前
力控面试题 01.01: 判定字符是否唯一(位运算) —— 题解
c++·学习·算法·leetcode·力控
wzdark3 天前
大规模并行计算中的负载均衡算法研究4
算法
吞下星星的少年·-·3 天前
C++ 萌新语法入门篇
c++·算法比赛