BZOJ 1898 沼泽鳄鱼(矩阵快速幂)

没有食人鱼不是裸题吗,用一个向量表示从s到1..N的距离,然后不停乘邻接矩阵行了,当然快速幂
有食人鱼,发现食人鱼最多十二个邻接矩阵一循环,处理出12个作为1个然后快速幂行了
 
怎么处理呢?
假设食人鱼在j时刻到达x这个点,那么j时刻的邻接矩阵x这一列全是0,因为他要求下一个矩阵的贡献上不能有x这一列贡献的。
 
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int N=52,MOD=1e4;
typedef long long ll;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}
int n,m,s,t,k,x,y,fish,f[15];
struct Mat{
    int a[N][N];
    Mat(){memset(a,0,sizeof(a));}
    void ini(){for(int i=1;i<=n;i++) a[i][i]=1;}
}g[15],ans;
inline Mat operator *(Mat A,Mat B){
    Mat C;
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++) if(A.a[i][k])
            for(int j=1;j<=n;j++) if(B.a[k][j])
                C.a[i][j]=(C.a[i][j]+A.a[i][k]*B.a[k][j])%MOD;
    return C;
}
inline Mat operator ^(Mat A,int k){
    Mat ans;ans.ini();
    for(;k;k>>=1,A=A*A)
        if(k&1) ans=ans*A;
    return ans;
}
void print(Mat A){
    puts("hiMat");
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++) printf("%d ",A.a[i][j]);
        puts("");
    }
}
int main(){
    //freopen("in.txt","r",stdin);
    n=read();m=read();s=read()+1;t=read()+1;k=read();
    for(int i=1;i<=m;i++){
        x=read()+1;y=read()+1;
        for(int j=1;j<=12;j++) g[j].a[x][y]=g[j].a[y][x]=1;
    }
    fish=read();
    for(int i=1;i<=fish;i++){
        int T=read();
        for(int j=1;j<=T;j++) f[j]=read()+1;
        for(int j=1;j<=12;j++)
            for(int k=1;k<=n;k++) g[j].a[k][f[j%T+1]]=0;
    }
    g[0].ini();
    for(int i=1;i<=12;i++) g[0]=g[0]*g[i];
    ans=g[0]^(k/12);
    for (int i=1;i<=k%12;i++) ans=ans*g[i];
    printf("%d",ans.a[s][t]);
}
View Code

 

posted @ 2017-04-03 15:13  free-loop  阅读(286)  评论(0编辑  收藏  举报