2021 ICPC 四川省赛 L - Spicy Restaurant（多源BFS，DP）

Spicy Restaurant

https://codeforces.com/gym/103117/problem/L

题目大意：给一个 $n$ 个点 $m$ 条边的点权图。再给出 $q$ 个询问，对于每个询问给出两个变量 $p,a$ ，问从 $p$ 开始走，到点权不超过 $a$ 的点的最短路是多少。

注意到这里的点权 $w$ 十分小，我们可以从这里下手考虑。假如我们设 $dp[i][j]$ 为从 $i$ 点出发，到点权不超过 $j$ 的点的最短路长度，我们会发现不好更新，有互相依赖的关系。那么我们不妨让 $j$ 定下来，设 $dp[i][j]$ 为从点 $i$ 出发，到达点权确切为 $j$ 的最短路。那么我们就可以对于每个 $j$ 计算答案，将 $w_i=j$ 的所有点作为源点去做多源 $BFS$ ，即可求得 $dp[i][j]$ 。那么我们发现我们可以轻松的转化为我们要求的答案，我们发现真正的答案 $ans[i][j]=min(dp[i][k])(k\le j)$ ，那么我们进行如下的代码的更新就可以线性得到答案：

for (int i = 1; i <= n; ++i) {
    for (int j = 2; j <= 100; ++j) {
        dp[i][j] = min(dp[i][j], dp[i][j - 1]);
    }
}

$code$

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10;

int n, m, q, w[N], dp[N][105], inf;

vector<int> g[N];

queue<int> qe;

void bfs(int x) {
    for (int i = 1; i <= n; ++i) {
        if (w[i] == x) {
            dp[i][x] = 0;
            qe.push(i);
        }
    }
    while(!qe.empty()) {
        int u = qe.front(); qe.pop();
        for (auto v : g[u]) {
            if (dp[v][x] == inf) {
                dp[v][x] = dp[u][x] + 1;
                qe.push(v);
            }
        }
    }
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    memset(dp, 0x3f, sizeof dp); inf = dp[0][0];
    scanf("%d%d%d", &n, &m, &q);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &w[i]);
    }
    while(m--) {
        int u, v; scanf("%d%d", &u, &v);
        g[u].emplace_back(v);
        g[v].emplace_back(u);
    }
    for (int i = 1; i <= 100; ++i) {
        bfs(i);
    }
    for (int i = 1; i <= n; ++i) {
        for (int j = 2; j <= 100; ++j) {
            dp[i][j] = min(dp[i][j], dp[i][j - 1]);
        }
    }
    while(q--) {
        int pos, t; scanf("%d%d", &pos, &t);
        printf("%d\n", dp[pos][t] == inf ? -1 : dp[pos][t]);
    }
}