Cow Relays(类矩阵快速幂优化dp)

题意翻译

给定一张 T 条边的无向连通图，求从 S 到 E 经过 N 条边的最短路长度。

输入格式

第一行四个正整数 N,T,S,E，意义如题面所示。

接下来 T 行每行三个正整数 w,u,v，分别表示路径的长度，起点和终点。

输出格式

一行一个整数表示图中从 S 到 E 经过 N 条边的最短路长度。

数据范围

2 <= T <= 100

2 <= N <= 1000000

思路： 采取dp状态转移的思想，令d[k][i][j]表示从i到j经过k条边的最短距离，状态转移方程为：d[a+b][i][j] = d[a][i][k] + d[b][k][j]，因此我们可以每次多走一条边求最短路，伪代码如下：

for (k: 1 ~ n)
    for (x: 1 ~ n)
        for (i: 1 ~ n)
            for (j: 1 ~ n)
                g[k][i][j] = min(g[k][i][j], g[k - 1][i][x] + g[1][x][j])

但是这么走的话太慢了，我们将状态转移方程写成d[a+b+c][i][j] = d[a][i][k] + d[b][k][x] + d[c][x][j]，式子右边三项互不影响，即该状态转移方程满足结合律，因此我们可以使用快速幂进行倍增加速。例如:⭣
d[8][i][j] = d[4][i][k] + d[4][k][j] = (d[2][i][x] + d[2][x][k]) + (d[2][k][y] + d[2][y][j]) =d[1][][] + ......
d[5][i][j] = d[4][i][k] + d[1][k][j] = d[2][i][x] + d[2][x][k] + d[1][k][j] = d[1][][] + ...

---

细节部分：
1.本题的边数最多为100，因此最多有200个有用的点，我们用map对点数进行离散化，赋予每个点唯一的编号
2.g[i][i]不用初始化为0，因为g数组的含义为各点走一步的最短路，而一个点走一步不可能到达自身
3.res[i][i]需要初始化为0，因为res数组的含义为各点之间走零步的最短路，这里的res类似快速幂中的1，一个数乘以1等于本身，因此一个最短距离多走零条边后还是本身不变
4.我们可以用类似数组覆盖的方式将dp降一维，比如g[][N]代表的是各点间走一步的最短路，我们用两个g求最短路，就可以让g更新为各点间走两步路的最短路:g=g*g

代码如下：

#include<bits/stdc++.h>
using namespace std;

#define N 130

map<int,int>mp;
int n,k,t,s,e;
int g[N][N],res[N][N];

void mul(int c[][N],int a[][N],int b[][N]){
    int temp[N][N];
    memset(temp,0x3f,sizeof temp);
    
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                temp[i][j]=min(temp[i][j],a[i][k]+b[k][j]);
    
    memcpy(c,temp,sizeof temp);
    
}

void qmi(){
    
    memset(res,0x3f,sizeof res);
    for(int i=1;i<=n;i++)res[i][i]=0;

    while(k){
        if(k&1)mul(res,res,g);//res=res*g
        k/=2;
        mul(g,g,g);//g=g*g
    }
}

int main(){
  
    memset(g,0x3f,sizeof g);
    
    cin>>k>>t>>s>>e;
    mp[s]=++n,mp[e]=++n;
    s=mp[s],e=mp[e];
    
    int a,b,c;
 
    while(t--){
        cin>>c>>a>>b;    
        if(!mp.count(a))mp[a]=++n;
        if(!mp.count(b))mp[b]=++n;        
        a=mp[a],b=mp[b];
        g[a][b]=g[b][a]=min(g[a][b],c);
    }
   
    qmi();
    
    printf("%d\n",res[s][e]);
    
    return 0;
}