【bzoj 1706】relays 奶牛接力跑（Floyd+快速幂）

传送门biu~
对于一个表示一条路最短路的矩阵和一个表示两条路最短路的矩阵，把它们Floyd合并，一定能合并出一个表示三条路最短路的矩阵。那么对于n条路的最短路，我们只需要把表示一条路最短路的矩阵合并n次就可以了，可以采用矩阵快速幂。

#include<bits/stdc++.h>
using namespace std;
int n,m,S,T;
int mp[1000005],cnt;
struct Matrix{
    int b[205][205];
    Matrix(){memset(b,0x3f,sizeof(b));}
    Matrix operator* (const Matrix&r)const{
        Matrix re;
        for(int k=1;k<=cnt;++k)
            for(int i=1;i<=cnt;++i)
                for(int j=1;j<=cnt;++j)
                    re.b[i][j]=min(re.b[i][j],b[i][k]+r.b[k][j]);
        return re;
    }
}a;
inline Matrix ksm(Matrix p,int n){
    Matrix re=p;
    --n;
    while(n){
        if(n&1) re=re*p;
        p=p*p;
        n>>=1;
    }
    return re;
}
int main(){
    scanf("%d%d%d%d",&n,&m,&S,&T);
    for(int i=1;i<=m;++i){
        int u,v,w;
        scanf("%d%d%d",&w,&u,&v);
        if(!mp[u])      mp[u]=++cnt;
        if(!mp[v])      mp[v]=++cnt;
        a.b[mp[u]][mp[v]]=a.b[mp[v]][mp[u]]=w;
    }
    printf("%d",ksm(a,n).b[mp[S]][mp[T]]);
    return 0;
}