最短路 2
Time Limit: 6000/4000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 417 Accepted Submission(s): 108
Problem Description
小 A 是社团里的工具人,有一天他的朋友给了他一个 n 个点,m 条边的正权连通无向图,要他计算所有点两两之间的最短路。
作为一个工具人,小 A 熟练掌握着 floyd 算法,设 w[i][j] 为原图中 (i,j) 之间的权值最小的边的权值,若没有边则 w[i][j]=无穷大。特别地,若 i=j,则 w[i][j]=0。
Floyd 的 C++ 实现如下:
```c++
for(int k=1;k<=p;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
w[i][j]=min(w[i][j],w[i][k]+w[k][j]);
```
当 p=n 时,该代码就是我们所熟知的 floyd,然而小 A 为了让代码跑的更快点,所以想减少 p 的值。
令 Di,j 为最小的非负整数 x,满足当 p=x 时,点 i 与点 j 之间的最短路被正确计算了。
现在你需要求 ∑ni=1∑nj=1Di,j,虽然答案不会很大,但为了显得本题像个计数题,你还是需要将答案对 998244353 取模后输出。
Input
第一行一个正整数 T(T≤30) 表示数据组数
对于每组数据:
第一行两个正整数 n,m(1≤n≤1000,m≤2000),表示点数和边数。
保证最多只有 5 组数据满足 max(n,m)>200
接下来 m 行,每行三个正整数 u,v,w 描述一条边权为 w 的边 (u,v),其中 1≤w≤109
Output
输出 T 行,第 i 行一个非负整数表示第 i 组数据的答案
Sample Input
1 4 4 1 2 1 2 3 1 3 4 1 4 1 1
Sample Output
6
假设一条最短路需要被floyd中更新,那么需要的最小k,就是这条最短路中经过的(不算起终点)最大的点的序号,那么我们对于每个起点i进行一次spfa选最短路的同时经过的点的最大序号最小就好了。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll inf = 1e15;
const int mod = 998244353;
const int N = 1010;
int head[N], cnt, n, m;
struct edge {
int u, v, ne, w;
}ed[N * 4];
struct node {
ll dis; int id;
}d[N];
void add(int u, int v, int w) {
ed[cnt].u = u; ed[cnt].v = v; ed[cnt].w = w;
ed[cnt].ne = head[u]; head[u] = cnt++;
}
void spfa(int st) {
for (int i = 1; i <= n; i++) {
d[i].dis = inf; d[i].id = 0;
}
d[st].dis = 0; d[st].id = 0;
queue<int>q;
for (int i = head[st]; ~i; i = ed[i].ne) {
int v = ed[i].v;
d[v].dis = min(1LL * ed[i].w, d[v].dis);
d[v].id = 0;
q.push(v);
}
while (!q.empty()) {
int u = q.front(); q.pop();
for (int i = head[u]; ~i; i = ed[i].ne) {
int v = ed[i].v; int laid;
if (d[v].dis > d[u].dis + 1LL * ed[i].w) {
d[v].dis = d[u].dis + 1LL * ed[i].w;
laid = max(d[u].id, u);
d[v].id = laid;
q.push(v);
}
else if (d[v].dis == d[u].dis + 1LL * ed[i].w) {
laid = max(d[u].id, u);
if (laid < d[v].id) {
d[v].id = laid;
q.push(v);
}
}
}
}
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0); cout.tie(0);
int te; cin >> te;
while (te--) {
cin >> n >> m;
memset(head, -1, sizeof head);
cnt = 0;
while (m--) {
int a, b, c;
cin >> a >> b >> c;
add(a, b, c); add(b, a, c);
}
ll sum = 0;
for (int i = 1; i <= n; i++) {
spfa(i);
for (int j = 1; j <= n; j++) {
sum += d[j].id;
}
}
sum %= mod;
cout << sum << "\n";
}
return 0;
}