题目
成都有n家火锅店,编号从1到n,第i家火锅店提供一定辣度的火锅。较高的辣度表示更辣的味道,而较低的辣度则更温和(但仍然需要非常小心)。 我们可以将这 n 家餐厅视为具有 m 条边的无向图上的节点。现在我们有q个游客想尝试一下火锅。给定游客当前的位置和他们所能承受的最大辣味值,你的任务是计算游客和他能接受的最近餐厅之间的最短距离。 在这个问题中,我们将路径的距离定义为路径中的边数。
题解
定义一个数组dis[i][j]
,表示火锅店j距离辣味值为i的火锅店的最小距离。
然后100次bfs即可。
疑惑点
bfs查找的过程。为啥不是这样的?dis[c][y] = min(dis[c][x] + 1, dis[c][y]);
而是这样的,dis[c][y] = dis[c][x]+1
。虽然两个都AC了。
疑惑的原因还是对搜索的过程不熟悉。搜索的时候是一层一层的来的,具体的说就是:先把辣味值为i结点入队,然后分别把辣味值为i的子结点入队(若已在队列中则跳过),然后再下一步就是子结点的的子结点,依次往复。所以可以求出j火锅店到i火锅店的最小距离(因为距离i火锅店近的,肯定先入队,先入队就先更新,所以可以做到最小)
#include<cstdio>
#include<queue>
#include<set>
#include<cstdlib>
#include<string.h>
#include<string>
#include<iostream>
#include<cmath>
#include<map>
#include<algorithm>
#define endl "\n"
#define IOS ios::sync_with_stdio(0), cin.tie(0),cout.tie(0)
#define ft first
#define sd second
#define pll pair<ll, ll>
#define pii pair<int, int>
#define ll long long int
#define mt(a,b) memset(a, b, sizeof a)
//#define int long long
const double PI = acos(-1.0);
const int inf = 0x3f3f3f3f;
const int INF = 0x7fffffff;
using namespace std;
const int N = 2e5 + 7, M = 1e6 + 10;
vector<int> v[N];
int w[N], dis[105][N];
int n;
queue<int> q;
void bfs(int c)
{
for (int i = 1; i <= n; i++)
if (w[i] == c) q.push(i), dis[c][i] = 0;
while (!q.empty())
{
int x = q.front(); q.pop();
for (int i = 0; i < v[x].size(); i++)
{
int y = v[x][i];
if (dis[c][y] != inf) continue;
q.push(y);
dis[c][y] = dis[c][x] + 1;
}
}
}
int main()
{
IOS;
int m, q;
cin >> n >> m >> q;
for (int i = 1; i <= n; i++) cin >> w[i];
for (int i = 1; i <= m; i++)
{
int x, y; cin >> x >> y;
v[x].push_back(y); v[y].push_back(x);
}
mt(dis, 0x3f);
for (int i = 1; i <= 100; i++)
bfs(i);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= 100; j++)
dis[j][i] = min(dis[j][i], dis[j - 1][i]);
while (q--)
{
int l, r; cin >> r >> l;
cout << (dis[l][r] == inf ? -1 : dis[l][r]) << endl;
}
return 0;
}