1706: [usaco2007 Nov]relays 奶牛接力跑

题目链接

题目大意：给定一张无向图，求从s出发恰好经过n条边到达e的最短路

题解：点编号比较大，离散化点，用f[i]表示经过i+1条边到达e的最短路，
，$f[i]=f[i-1]*f[i-2]，f[k-1]=f[0]^k$ ， 大概跟优化线性递推差不多
因为修改了矩阵乘法的定义，单位矩阵也应该修改，直接手动乘一次就好

我的收获：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define INF 1010580540
using namespace std;

const int M=205;

int n,K,s,e,m,cnt,mp[1005];

struct Mat{int x,m[M][M];Mat(){x=1,memset(m,0x3f,sizeof(m));}}w;

int ID(int x){return mp[x]?mp[x]:mp[x]=++cnt;}

Mat operator*(Mat a,Mat b){
    Mat c;c.x=a.x;
    for(int i=1;i<=a.x;i++)
        for(int j=1;j<=b.x;j++)
            for(int k=1;k<=a.x;k++)
                c.m[i][j]=min(c.m[i][j],a.m[i][k]+b.m[k][j]);
    return c;
}

Mat operator^(Mat a,int p){
    Mat c;bool flag=true;
    for(;p;p>>=1,a=a*a)
    if(p&1) {if(flag) c=a,flag=false;else c=c*a;}
    return c;
}

void work()
{
    Mat ans=w^K;
    printf("%d",ans.m[s][e]);
}

void init()
{
    int x,y,z;
    scanf("%d%d%d%d",&K,&m,&s,&e);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&z,&x,&y);
        x=ID(x);y=ID(y);
        w.m[x][y]=w.m[y][x]=z;
    }
    s=ID(s);e=ID(e);w.x=cnt;
}

int main()
{
    init();
    work();
    return 0;
}