【poj3613】cow relays【floyd快速幂】

传送门

其实还是矩阵快速幂。将图用邻接矩阵表示，这个矩阵自乘多少次，得到的结果就是两点间经过(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;
}