bzoj1898 [Zjoi2005]Swamp 沼泽鳄鱼——矩阵快速幂

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=1898

求到达方案数...还是矩阵快速幂；

能够到达就把邻接矩阵那里赋成1，有鳄鱼的地方从各处来的路径都是0；

因为时间周期只有2，3，4，所以每12个时间就是一个循环；

可以用矩阵快速幂做了。

一开始写了个一团糟，也不明白样例为什么输出0了...

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
int const mod=;
int n,m,st,ed,K,head[],ct,nf,ey[][],te[],ans[],tmp[];
struct E{
    int to,next;
    E(int t=,int n=):to(t),next(n) {}
}edge[];
struct Matrix{
    int a[][];
    Matrix operator * (const Matrix &y) const
    {
        Matrix x;
        memset(x.a,,sizeof x.a);
        for(int i=;i<=n;i++)
            for(int k=;k<=n;k++)
                for(int j=;j<=n;j++)
                    (x.a[i][j]+=a[i][k]*y.a[k][j])%=mod;
        return x;
    }
}s[],b;
void add(int x,int y){edge[++ct]=E(y,head[x]); head[x]=ct;}
void build1()
{
    for(int x=;x<=n;x++)
        for(int i=;i<=;i++)
            for(int j=head[x],u;j;j=edge[j].next)
                s[i].a[x][u=edge[j].to]=;//
}
void build2()
{
    for(int i=;i<=nf;i++)
        for(int t=;t<=;t++)
            for(int j=;j<=n;j++)
            {
                int x=ey[i][t%te[i]];
    //            for(int k=head[x],y;k;k=edge[k].next)
    //                s[j].a[y=edge[k].to][x]=0;
                s[t].a[j][x]=;
            }
}
//void init(Matrix &x)
//{
//    memset(x.a,0,sizeof x.a);
//    for(int i=1;i<=55;i++)
//        for(int j=55;j;j--)
//            x.a[i][j]=1;
//}
void mul2(int a[],int b[][],int c[])
{
    int tmp[];
        for(int j=;j<n;j++)
        {
            tmp[j]=;
            for(int k=;k<n;k++)
               tmp[j]=(tmp[j]+a[k]*b[k][j])%mod;
        }
        for(int j=;j<n;j++)
            c[j]=tmp[j];
}
int main()
{
    scanf("%d%d%d%d%d",&n,&m,&st,&ed,&K);
    st++; ed++;
    for(int i=,x,y;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        x++; y++;
        add(x,y); add(y,x);
    }
    scanf("%d",&nf);
    for(int i=;i<=nf;i++)
    {
        scanf("%d",&te[i]);
        for(int j=;j<=te[i];j++)
            scanf("%d",&ey[i][j]),ey[i][j]++;
    }
    build1(); build2();
//    init(s[13]);
    for(int i=;i<=n;i++)
        b.a[i][i]=s[].a[i][i]=;
    for(int i=;i<=;i++)s[]=s[]*s[i];
//    Matrix b; init(b);
    int t=K/;
    while(t)
    {
        if(t&)b=b*s[];
        s[]=s[]*s[];
        t>>=;
    }
    t=K%;
    for(int i=;i<=t;i++)b=b*s[i];
//    for(int i=1;i<=55;i++)
//        for(int j=1;j<=55;j++)
//            if(b.a[i][j])printf("b:%d\n",b.a[i][j]);
//    tmp[st]=1;
//    for(int i=1;i<=n;i++)
//        for(int j=1;j<=n;j++)
//            (ans[i]+=tmp[j]*b.a[j][i])%=mod;
    ans[st]=;
    mul2(ans,b.a,ans);
    printf("%d",ans[ed]);
    return ;
}

囧

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
int const mod=;
int n,m,st,ed,K,nf,ey[];
struct Matrix{
    int a[][];
    Matrix operator * (const Matrix &y) const
    {
        Matrix x;
        memset(x.a,,sizeof x.a);
        for(int i=;i<=n;i++)
            for(int k=;k<=n;k++)
                for(int j=;j<=n;j++)
                    (x.a[i][j]+=a[i][k]*y.a[k][j])%=mod;
        return x;
    }
    Matrix(){memset(a,,sizeof a);}//
    void init()
    {
        for(int i=;i<=n;i++) a[i][i]=;
    }
}s[],b;
Matrix operator ^ (Matrix a,int k)
{
    Matrix ret; ret.init();
    while(k)
    {
        if(k&)ret=ret*a;
        a=a*a;
        k>>=;
    }
    return ret;
}
int main()
{
    scanf("%d%d%d%d%d",&n,&m,&st,&ed,&K);
    st++; ed++;
    for(int i=,x,y;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        x++; y++;
        for(int j=;j<=;j++)
            s[j].a[x][y]=s[j].a[y][x]=;
    }
    scanf("%d",&nf);
    for(int i=,tt;i<=nf;i++)
    {
        scanf("%d",&tt);
        for(int j=;j<=tt;j++)
            scanf("%d",&ey[j]),ey[j]++;
        for(int j=;j<=;j++)
            for(int k=;k<=n;k++)
                s[j].a[k][ey[j%tt+]]=;//+1
    }
    s[].init();
    for(int i=;i<=;i++)s[]=s[]*s[i];
    b=s[]^(K/);
    for(int i=;i<=K%;i++)b=b*s[i];
    printf("%d",b.a[st][ed]);
    return ;
}