floyd算法思路:
floyd思路其实就是dp,引入k,i,j,3个变量,k代表经过的边数,i,j,代表从i点到j点
采用dp思想,就有d[k,i,j]=d[k-1,i,k]+d[k-1,k,j]。还可以采用背包简化思想,优化空间,有
d[i][j]=d[i][k]+d[k][j]。
模板:
模板题:
题目描述
给出一张由 n 个点 m 条边组成的无向图。
求出所有点对 (i,j) 之间的最短路径。
输入格式
第一行为两个整数 n,m,分别代表点的个数和边的条数。
接下来 m 行,每行三个整数 u,v,w,代表 u,v 之间存在一条边权为 w 的边。
输出格式
输出 n 行每行 n 个整数。
第 i 行的第 j 个整数代表从 i 到 j 的最短路径。
代码:
#define _CRT_SECURE_NO_WARNINGS
#include<bits/stdc++.h>
using namespace std;
const int N = 110;
int d[N][N];
int n, m;
int main()
{
scanf("%d%d", &n, &m);
memset(d, 0x3f, sizeof(d));
for (int i = 1;i <= m;i++) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
d[a][b] = min(c, d[a][b]);
d[b][a] = min(c, d[b][a]);
}
for (int i = 1;i <= n;i++) {
d[i][i] = 0;
}
for (int k = 1;k <= n;k++) {
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++) {
d[i][j] = min(d[i][k] + d[k][j], d[i][j]);
}
}
}
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++) {
printf("%d ", d[i][j]);
}
printf("\n");
}
}
唯一注意的是要注意重复边取最小值。
应用:
例一:acwing:3512. 最短距离总和
给定一张带权无向完全图,设点的编号为 1,2,3,4,5....n(以邻接矩阵的形式给出)。
计算依次拿走第 i个点后,剩余所有点到其他点的最短距离之和的总和(具体请看样例)。
输入格式
第一行包含一个整数 n。
接下来 n 行,每行包含 n 个空格隔开的整数,表示邻接矩阵,其中第 i 行第 j 列的数字 aij 表示点 i 和点 j 之间存在一条边,长度为 aij。
输出格式
一个整数,表示最短距离之和的总和。
数据范围
1≤n≤500,
aii=0,
0<aij≤10 (i≠j)
这题需要理解floyd的dp思想,每拿掉一次用一次floyd显然会tle,所以考虑用一次,从末尾开始,逐渐加点。(和并查集的逆用有点像)
代码:
#define _CRT_SECURE_NO_WARNINGS
#include<bits/stdc++.h>
using namespace std;
const int N = 510;
int d[N][N];
int n;
int main()
{
scanf("%d", &n);
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++) {
scanf("%d", &d[i][j]);
}
}
int res = 0;
for (int k = n;k >= 2;k--) {
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++) {
d[i][j] = min(d[i][k] + d[k][j], d[i][j]);
if (i >= k && j >= k) {
res += d[i][j];
}
}
}
}
printf("%d", res);
}
例二:acwing:4872. 最短路之和
这题基本和上题一致,但更灵活。
代码:
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<unordered_set>
using namespace std;
typedef long long LL;
const int N = 510;
int d[N][N];
int th[N];
int n;
bool st[N];
LL ans[N];
int main()
{
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
scanf("%d", &d[i][j]);
}
}
for (int i = 1; i <= n; i++) {
scanf("%d", &th[i]);
}
for (int k = n ; k >= 1; k--) {
LL res = 0;
st[th[k]]=true;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
d[i][j] = min(d[i][th[k]] + d[th[k]][j], d[i][j]);
}
}
for (int i = 1; i <= n; i++) {
if(!st[i]){
continue;
}
for (int j = i+1; j <= n; j++) {
if (st[j]) {
res += d[i][j]+d[j][i] ;
}
}
}
ans[k] = res;
}
for (int i = 1; i <= n; i++) {
printf("%lld ", ans[i]);
}
}