link

題目大意

給你一個含有邊權的無向圖，問從$S$到$T$經過$N$條邊的最小花費。

試題分析

我們可以很容易推導$dp$方程,$dp(k,i,j)$表示經過$k$條邊從$i$到$j$的最小花費。則，$dp(k,i,j)=min(dp(k-1,i,p)+dp(1,p,j))$。

而$(i,p),(p,j),(i,j)$發現了什麼，這不是矩陣嗎，$dp(1,i,j)$為初始矩陣($1$次冪),$dp(2,i,j)$為$2$次冪，$dp(3,i,j)$為$3$次冪，所以只需要矩陣快速冪一下即可。

#include<iostream>
#include
 
<cstring>
#include<cstdio>
#include<algorithm>
#define int long long
using namespace std;
inline int read(){
    int f=1,ans=0;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+c-'0';c=getchar();}
    
 
return f*ans;
}
const int MAXN=101;
struct matrix{
    int st[MAXN][MAXN];
}a,F,ans;
struct node{
    int u,v,w;
}x[MAXN];
int Map[1000001],n,k,m,S,T,cnt;
matrix mul(matrix s1,matrix s2){
    matrix s3;
    memset(s3.st,127/3,sizeof(s3.st));
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            
 
for(int p=1;p<=n;p++){
                s3.st[i][p]=min(s3.st[i][p],s1.st[i][j]+s2.st[j][p]);
            }
    return s3;
}
matrix qpow(int b){
    if(b==0) return a;
    ans=a;
    while(b){
        if(b&1) ans=mul(ans,a);
        a=mul(a,a);b>>=1;
    }return ans;
}
signed main(){
    k=read(),m=read(),S=read(),T=read();
    memset(a.st,127/3,sizeof(a.st));
    for(int i=1;i<=m;i++){
        int w=read(),u=read(),v=read();
        if(Map[u]==0) Map[u]=++cnt;
        if(Map[v]==0) Map[v]=++cnt;
        u=Map[u],v=Map[v];
        x[i].w=w,x[i].u=u,x[i].v=v;
        a.st[u][v]=a.st[v][u]=min(a.st[u][v],w);
    }
    n=cnt;
    F=qpow(k-1);
    printf("%lld",F.st[Map[S]][Map[T]]);
}
/*
2 3 1 3
1 1 2
1 2 3
1 1 3
*/