[SCOI2009]BZOJ1297 迷路 - 矩阵乘法

传送门

题目大意：自行参考。

题解：如果边权都是1那么这个题是傻逼题，邻接矩阵自乘若干次即可。

显然可以每条边拆点但是复杂度不对。

考虑把时间花在路上（也就是对边拆点）等价于一个单位时间到达，但是“为了进入这个点”又花费了边权-1的时间。

因此对每个点i拆点，拆成10个，对于第i个点拆出来的第j个记作P(i,j)，到达P(i,1)才算到i这个点；

边(i,j,k)改为(P(i,1),P(j,k),1)。这样就可以把矩阵写大一点然后自乘了。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 2009
#define P(i,k) ((i-1)*x+k)
#define MAXA 110
using namespace std;
struct matrix{
    int v[MAXA][MAXA],n,m;
    matrix(int _n=0,int _m=0)
    {
        set_size(_n,_m);
    }
    inline int set_size(int _n=0,int _m=0)
    {
        n=_n,m=_m;
        for(int i=1;i<=n;i++)
            memset(v,0,sizeof(v));
        return 0;
    }
    inline matrix operator=(const matrix &b)
    {
        n=b.n,m=b.m;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                v[i][j]=b.v[i][j];
        return *this;
    }
    inline matrix operator*(const matrix &b)
    {
        matrix &a=*this,c(a.n,b.m);
        if(a.m^b.n) return c;
        for(int i=1;i<=a.n;i++)
            for(int k=1;k<=b.m;k++)
                for(int j=1;j<=a.m;j++)
                    c.v[i][k]=(c.v[i][k]+a.v[i][j]*b.v[j][k])%mod;
        return c;
    }
    friend matrix operator*=(matrix &a,const matrix &b)
    {
        return a=a*b;
    }
};
inline matrix fast_pow(const matrix &b,int k)
{
    if(k==1) return b;
    matrix ans=fast_pow(b,k>>1);ans*=ans;
    return (k&1)?(ans*b):ans;
}
int main()
{
    int n,t,x=9;scanf("%d%d",&n,&t);
    matrix B(x*n,x*n);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            int k;scanf("%1d",&k);
            if(k) B.v[P(i,1)][P(j,k)]=1;
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<x;j++)
            B.v[P(i,j+1)][P(i,j)]=1;
    B=fast_pow(B,t);
    printf("%d\n",B.v[P(1,1)][P(n,1)]);
    return 0;
}