题目描述
输入第一行给出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;
}