重返天梯-L3-008 喊山 (30 分)(bfs)

最新推荐文章于 2023-09-08 16:53:30 发布

CodeSlogan

最新推荐文章于 2023-09-08 16:53:30 发布

阅读量361

点赞数

CC 4.0 BY-SA版权

分类专栏：算法题文章标签： c++ bfs 天梯赛

本文链接：https://blog.youkuaiyun.com/comscience/article/details/124126597

算法题专栏收录该内容

70 篇文章

订阅专栏

该博客介绍了一个使用C++编写的简单广度优先搜索（BFS）算法，用于解决给定山头间声音可及性关系，查询给定山头能听到的最远山头的问题。代码中通过邻接列表存储山头之间的关系，并利用队列进行BFS遍历，找出最远距离的山头。

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

题目描述

输入第一行给出3个正整数n、m和k，其中n（≤10000）是总的山头数（于是假设每个山头从1到n编号）。

接下来的m行，每行给出2个不超过n的正整数，数字间用空格分开，分别代表可以听到彼此的两个山头的编号。这里保证每一对山头只被输入一次，不会有重复的关系输入。

最后一行给出k（≤10）个不超过n的正整数，数字间用空格分开，代表需要查询的山头的编号。

C++

很简单的bfs，就不多说了，直接看代码

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
typedef pair<int, int> PII;
const int N = 1e4 + 10, M = 2 * N;
int n, m, k;
int h[N], e[M], ne[M], idx;
int dist[N];
bool st[N];

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

int bfs(int x) {
	// 每次都要记得还原
    memset(dist, 0x3f, sizeof dist);
    memset(st, false, sizeof st);
    queue<PII> q;
    q.push({0, x});
    dist[x] = 0;
    
    int maxd = 0; // 记录最远距离
    
    while (q.size()) {
        auto t = q.front();
        q.pop();
        
        int ver = t.second, distance = t.first;
        
        if (st[ver]) continue;
        st[ver] = true;
        maxd = max(maxd, distance);
        dist[ver] = distance;
        for (int i = h[ver]; ~i; i = ne[i]) {
            int j = e[i];
            
            if (!st[j]) {
                q.push({distance + 1, j});
            }
        }
    }
    
    for (int i = 1; i <= n; i++) {
        if (dist[i] == maxd) {
            maxd = i;
            break;
        }
    }
    if (maxd == x) return 0;
    else return maxd;
}


int main() {
    scanf("%d%d%d", &n, &m, &k);
    memset(h, -1, sizeof h);
    for (int i = 0; i < m; i++) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }
    
    while (k--) {
        int x;
        scanf("%d", &x);
        cout << bfs(x) << endl;
    }
    
    return 0;
}