bzoj1297 / P4159 [SCOI2009]迷路

P4159 [SCOI2009]迷路

如果边权只有 0/1 那么不就是一个灰常简单的矩阵快速幂吗！

然鹅边权 $<=9$

所以我们把每个点拆成9个点！

解决~

#include<iostream>
#include<cstdio>
#include<cstring>
#define re register
using namespace std;
const int mod=2009;
int m,n,t;long long q;
struct matrix{
    int a[92][92];
    matrix(){memset(a,0,sizeof(a));}
    matrix operator * (const matrix &tmp) const{
        matrix c;
        for(int i=1;i<=n;++i)
            for(int j=1;j<=n;++j)
                for(int k=1;k<=n;++k)
                    c.a[i][j]=(c.a[i][j]+a[i][k]*tmp.a[k][j])%mod;
        return c;
    }
    matrix Pow(matrix x,int y){
        matrix res;
        for(int i=1;i<=n;++i) res.a[i][i]=1;
        for(;y;y>>=1,x=x*x)
            if(y&1) res=res*x;
        return res;
    }
}st;
int idx(int x,int y){return x+m*y;}//新点编号
int main(){
    scanf("%d%d",&m,&t);n=m*9;
    for(int i=1;i<=m;++i){
        for(int j=1;j<=8;++j)
            st.a[idx(i,j)][idx(i,j-1)]=1;//规定idx(i,0)作为原图的点向其他点连边，其他点与该点的边权就转化为连到idx(i,dist-1)上
        scanf("%lld",&q);
        for(int j=m;j>=1;--j,q/=10)
            if(q%10) st.a[i][idx(j,q%10-1)]=1;
    }st=st.Pow(st,t);
    printf("%d",st.a[1][m]);
    return 0;
}