【刷爆LeetCode】五月算法集训（17）广度优先搜索

最新推荐文章于 2024-09-15 15:43:59 发布

...1999

最新推荐文章于 2024-09-15 15:43:59 发布

阅读量176

点赞数

分类专栏：五月算法集训文章标签： leetcode 算法宽度优先

本文链接：https://blog.csdn.net/LAdestiny/article/details/125753875

版权

五月算法集训专栏收录该内容

32 篇文章 1 订阅

订阅专栏

本文介绍了四个算法题目，包括LCP44开幕式焰火（使用哈希表和队列实现）、102二叉树的层序遍历、1609奇偶树及1263推箱子。通过C++代码展示了广度优先搜索在解决树形问题中的应用，适合提升算法能力。

摘要由CSDN通过智能技术生成

题目来源于知识星球—英雄算法联盟，五月算法集训专题

前言

跟随英雄算法联盟博主—英雄哪里出来，每天完成相应的算法练习，一个月后，必定会有所成长！

一、LCP 44.开幕式焰火（简单）

1.题目描述

2.解题思路

用到了哈希表和队列。将根节点放入队列，利用广度优先搜索，将每个扩展出来
的节点放入哈希表，放入之前统计是否出现过并计数。

3.代码演示（C++）

class Solution 
{
public:
    int numColor(TreeNode* root) 
    {
        int hash[1001],cnt=0;
        queue<TreeNode*>q;
        memset(hash,0,sizeof(hash));
        q.push(root);
        while(!q.empty())
        {
            TreeNode *now=q.front();
            q.pop();
            if(now)
            {
                if(!hash[now->val])
                {
                    hash[now->val]=1;
                    ++cnt;
                }
                q.push(now->left);
                q.push(now->right);
            }
        }
        return cnt;
    }
};

4.题目链接

题目传送门

二、102.二叉树的层序遍历（中等）

1.题目描述

2.解题思路

将根节点放入队列，入队时记录深度，出队时，根据深度来选择放入结果
数组的哪一层，之后套用广度优先搜索的模板。

3.代码演示（C++）

class Solution 
{
public:
    vector<vector<int>> levelOrder(TreeNode* root) 
    {
        if(root==nullptr)
        {
            return {};
        }
        vector<vector<int>>ret;
        unordered_map<TreeNode*,int>dep;
        queue<TreeNode*>q;
        dep[root]=0;
        int predep=-1;
        q.push(root);
        while(!q.empty())
        {
            TreeNode* n=q.front();
            q.pop();
            if(dep[n]!=predep)
            {
                ret.push_back({n->val});
                predep=dep[n];
            }
            else 
            {
                ret.back().push_back(n->val);
            }
            if(n->left)
            {
            dep[n->left]=dep[n]+1;
            q.push(n->left);
            }
            if(n->right)
            {
            dep[n->right]=dep[n]+1;
            q.push(n->right);
            }
        } 
        return ret;
    }
    
};