给定一张由T条边构成的无向图,点的编号为1~1000之间的整数。
求从起点S到终点E恰好经过N条边(可以重复经过)的最短路。
注意: 数据保证一定有解。
输入格式
第1行:包含四个整数N,T,S,E。
第2…T+1行:每行包含三个整数,描述一条边的边长以及构成边的两个点的编号。
输出格式
输出一个整数,表示最短路的长度。
数据范围
2≤T≤100,
2≤N≤106
输入样例:
2 6 6 4
11 4 6
4 4 8
8 4 9
6 6 8
2 6 9
3 8 9
输出样例:
10
心态炸裂,不知道是网站的问题还是我代码的问题,结构体和全局变量的定义顺序反一下,答案就有错的,改了快一个小时。。。
思路
bellmanford 上次用的是去求不超过k条边的最短路,这个是恰好是k条边的最短路,所以需要改一下下,如果我们每次让它强制更新的话就会变成求解恰好经过k条边的最短路。
memset(dis, 0x3f, sizeof dis);
dis[s] = 0;
for(register int i = 1; i <= n; i++){
memcpy(bDis, dis, sizeof dis);
memset(dis, 0x3f, sizeof dis);每次把点的距离初始化成0x3f那么它每次都会被转移
for(register int j = 1; j <= m; j++){
dis[e[j].v] = min(dis[e[j].v], bDis[e[j].u] + e[j].w);
dis[e[j].u] = min(dis[e[j].u], bDis[e[j].v] + e[j].w);
}
}
#pragma GCC optimize(2)
#include <bits/stdc++.h>
using namespace std;
const int N = 205, M = 105;
struct kl{
int u,v,w;
}e[505];
int m, n, s, t, adj[N], dis[N], bDis[N], tot,st[510],id[510],idx;
int inline bellmanFord(){
memset(dis, 0x3f, sizeof dis);
dis[s] = 0;
for(register int i = 1; i <= n; i++){
memcpy(bDis, dis, sizeof dis);
memset(dis, 0x3f, sizeof dis);
for(register int j = 1; j <= m; j++){
dis[e[j].v] = min(dis[e[j].v], bDis[e[j].u] + e[j].w);
dis[e[j].u] = min(dis[e[j].u], bDis[e[j].v] + e[j].w);
}
}
return dis[t];
}
int main(){
cin>>n>>m>>s>>t;
if(!st[s]) id[s]=++idx,st[s]=1;
if(!st[t]) id[t]=++idx,st[t]=1;
s=id[s],t=id[t];
for(int i=1;i<=m;i++){
int a1,b,c; cin>>a1>>b>>c;
if(st[b]==0) id[b]=++idx,st[b]=1;
if(st[c]==0) id[c]=++idx,st[c]=1;
b=id[b],c=id[c];
e[i].u=b,e[i].v=c,e[i].w=a1;
}
printf("%d\n", bellmanFord());
return 0;
}