51nod3225 小明和最短路

3225 小明和最短路

经过一段时间的修炼，小明已经成为最短路大师，最近他设计了这样一个问题。给定一个图，小明想要知道从s点到e点正好经过图里面k条边的最短路的长度是多少，聪明的你可以帮助小明解决这个问题吗？

输入

第一行输入两个整数n和k；
接下来n-1行，每行包含两个数x，y和w，表示x和y直接有一条长度为w的路径；
接下来一行输入两个整数s和e，表示起点和终点。

输出

输出一个整数，表示从s点到e点正好经过图里面k条边的最短路的长度，如果不存在这样一条道路，输出-1。

数据范围

对于10%的数据，1<= n<= 4， 1<=k<=8；
对于50%的数据，1<= n<= 100， 1<=k<=2048；
对于100%的数据，1<= n<=100，1<=k<= 1000000；
1<=w<=100。

输入样例

3 2 4
1 2 3
2 3 4
1 3

输出样例

13

解析：

这个题目是一个技巧题，解法为矩阵快速幂+floyd，具体做法就是把矩阵快速幂里面改成floyd的计算即可，每次可以增加一条边，最后得到的结果保证经过了k条边，需要注意的是无解的情况，特判一下即可，时间复杂度O(n^3logk)。

放代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int inf = 0x3f3f3f3f;
int read(){
	int x=0,f=0;char ch=getchar();
	while(!isdigit(ch)) {if(ch == '-')f=1;ch=getchar();}
	while(isdigit(ch)) {x=x*10+ch-'0';ch=getchar();}
	return f ? -x : x ;
}
const int N = 110;
int f[N][N],n,k,dp[N][N],m,res[N][N];
int main()
{
	memset(dp,0x3f,sizeof(dp));
	n = read();m = read();k = read();
	for(int i = 1;i <= m;i++) {
		int a = read(),b = read(),c = read();
		dp[a][b] = min(dp[a][b],c);
		dp[b][a] = min(dp[b][a],c);
	}
	int s = read(),t = read();
	memset(f,0x3f,sizeof(f));
	f[s][s] = 0;
	while(k) {
		if(k&1) {
			memset(res,0x3f,sizeof(res));
			for(int i = 1;i <= n;i++){
				for(int j = 1;j <= n;j++){
					for(int k = 1;k <= n;k++){
						res[i][j] = min(res[i][j],f[i][k]+dp[k][j]);
					}
				}
			}
			swap(res,f);
		}
		k >>= 1;
		memset(res,0x3f,sizeof(res));
		for(int i = 1;i <= n;i++){
			for(int j = 1;j <= n;j++){
				for(int k = 1;k <= n;k++){
					res[i][j] = min(res[i][j],dp[i][k] + dp[k][j]);
				}
			}
		}
		swap(res,dp);
	}
	if(f[s][t] == 0x3f3f3f3f) cout <<"-1"<<endl;
	else cout << f[s][t] << endl;
	return 0;
}