题意: 求进过路径数量为k的已知起点的最短路 ;
先进行在线离散化。
然后Floyd 矩阵快速幂。
注意 如果结构体中数组大小开太大 (大概极限是350乘350)的话不能动态使用,就是将结构体导入导出函数,需要在外部声明一个常量的结构体,对其进行改变。
//#include <bits/stdc++.h>
#include <map>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
using namespace std ;
const int inf = 0x3f3f3f3f ;
map<int , int> pq ;
int pq_num ;
struct Matrix{
int m[250][250] ;
void init(){
memset(m , 0x3f , sizeof(inf)) ;
}
Matrix operator * (const Matrix & temp) const {
Matrix c ;
memset(c.m , inf , sizeof(c.m)) ;
for(int i = 1 ; i < pq_num ; i ++ ){
for(int j = 1 ; j < pq_num ; j ++ ){
for(int k = 1 ; k < pq_num ; k ++ )
c.m[i][j] = min(c.m[i][j] , m[i][k] + temp.m[k][j]) ;
}
}
return c ;
}
};
Matrix ans , read ;
void power(int mi){
//Matrix ans ; ans.init() ;
//n -- ;
ans = read ;
mi -- ;
while( mi ){
if(mi & 1){
ans = ans * read ;
}
mi >>= 1 ;
//base = base * base ;
read = read * read ;
}
//return ans ;
}
int main(){
int n , T , st , ed , a , b , w;
//cout << "fsagfads" ;
scanf("%d %d %d %d" , &n , &T , &st , &ed) ;
pq.clear() ;
pq_num = 1 ;
///read.init() ;
//cout << read.m[0][0] << endl ;
memset(read.m , 0x3f , sizeof(read.m)) ;
//cout << read.m[0][0] << endl ;
//cout << read.m[0][0] << endl ;
for(int i = 0 ; i < T ; i ++ ){
scanf("%d %d %d" , &w , &a , &b) ;
if(!pq[a]) pq[a] = pq_num ++ ;
if(!pq[b]) pq[b] = pq_num ++ ;
a = pq[a] , b = pq[b] ;
read.m[a][b] = read.m[b][a] = w ;
}
st = pq[st] , ed = pq[ed] ;
//power(read , n) ;
power(n) ;
printf("%d\n" , ans.m[st][ed]) ;
//}
return 0 ;
}