其实还是矩阵快速幂。将图用邻接矩阵表示,这个矩阵自乘多少次,得到的结果就是两点间经过(n+1)条边的最短路。
记得每次都要memset成inf。我在ri里面进行了(日)
顺带一提,开long long会TLE。。(鈤)
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
#define in read()
#define int long long
int in{
int cnt=0,f=1;char ch=0;
while(!isdigit(ch)){
ch=getchar();if(ch=='-')f=-1;
}
while(isdigit(ch)){
cnt=cnt*10+ch-48;
ch=getchar();
}return cnt*f;
}
int n,t,s,e;
int q[1003],cnt,vis[1003];
int ans[1003][1003],now[1003][1003],dis[1003][1003];
int gu[1003][1003];
void floyd(int a[][1003],int b[][1003],int c[][1003]){
for(int i=1;i<=cnt;i++){
for(int j=1;j<=cnt;j++){
for(int k=1;k<=cnt;k++){
a[q[j]][q[k]]=min(a[q[j]][q[k]],b[q[i]][q[j]]+c[q[i]][q[k]]);
}
}
}
}
void ri(int a[][1003],int b[][1003]){
for(int i=1;i<=cnt;i++){
for(int j=1;j<=cnt;j++){
a[q[i]][q[j]]=b[q[i]][q[j]];b[q[i]][q[j]]=0x3f3f3f3f;
}
}
}
void ksm(int b){
while(b){
if(b&1){
floyd(dis,ans,gu);
ri(ans,dis);
}
floyd(now,gu,gu);
ri(gu,now);b>>=1;
}
}
signed main(){
n=in;t=in;s=in;e=in;
memset(dis,0x3f,sizeof(dis));
memset(now,0x3f,sizeof(now));
memset(ans,0x3f,sizeof(ans));
memset(gu,0x3f,sizeof(gu));
for(int i=1;i<=t;i++){
int a=in;int b=in;int c=in;
gu[b][c]=gu[c][b]=min(gu[b][c],a);
if(!vis[b]){vis[b]=1;q[++cnt]=b;}
if(!vis[c]){vis[c]=1;q[++cnt]=c;}
}
for(int i=1;i<=1000;i++)ans[i][i]=0;
ksm(n);
cout<<ans[s][e];
return 0;
}