bzoj 1898: [Zjoi2005]Swamp 沼泽鳄鱼

Description
潘塔纳尔沼泽地号称世界上最大的一块湿地，它地位于巴西中部马托格罗索州的南部地区。每当雨季来临，这里碧波荡漾、生机盎然，引来不少游客。为了让游玩更有情趣，人们在池塘的中央建设了几座石墩和石桥，每座石桥连接着两座石墩，且每两座石墩之间至多只有一座石桥。这个景点造好之后一直没敢对外开放，原因是池塘里有不少危险的食人鱼。豆豆先生酷爱冒险，他一听说这个消息，立马赶到了池塘，想做第一个在桥上旅游的人。虽说豆豆爱冒险，但也不敢拿自己的性命开玩笑，于是他开始了仔细的实地勘察，并得到了一些惊人的结论：食人鱼的行进路线有周期性，这个周期只可能是2，3或者4个单位时间。每个单位时间里，食人鱼可以从一个石墩游到另一个石墩。每到一个石墩，如果上面有人它就会实施攻击，否则继续它的周期运动。如果没有到石墩，它是不会攻击人的。借助先进的仪器，豆豆很快就摸清了所有食人鱼的运动规律，他要开始设计自己的行动路线了。每个单位时间里，他只可以沿着石桥从一个石墩走到另一个石墩，而不可以停在某座石墩上不动，因为站着不动还会有其它危险。如果豆豆和某条食人鱼在同一时刻到达了某座石墩，就会遭到食人鱼的袭击，他当然不希望发生这样的事情。现在豆豆已经选好了两座石墩Start和End，他想从Start出发，经过K个单位时间后恰好站在石墩End上。假设石墩可以重复经过（包括Start和End），他想请你帮忙算算，这样的路线共有多少种（当然不能遭到食人鱼的攻击）。

Input
输入文件共M + 2 + NFish行。第一行包含五个正整数N，M，Start，End和K，分别表示石墩数目、石桥数目、Start石墩和End石墩的编号和一条路线所需的单位时间。石墩用0到N–1的整数编号。第2到M + 1行，给出石桥的相关信息。每行两个整数x和y，0 ≤ x, y ≤ N–1，表示这座石桥连接着编号为x和y的两座石墩。第M + 2行是一个整数NFish，表示食人鱼的数目。第M + 3到M + 2 + NFish行，每行给出一条食人鱼的相关信息。每行的第一个整数是T，T = 2，3或4，表示食人鱼的运动周期。接下来有T个数，表示一个周期内食人鱼的行进路线。 如果T＝2，接下来有2个数P0和P1，食人鱼从P0到P1，从P1到P0，……； 如果T＝3，接下来有3个数P0，P1和P2，食人鱼从P0到P1，从P1到P2，从P2到P0，……； 如果T＝4，接下来有4个数P0，P1，P2和P3，食人鱼从P0到P1，从P1到P2，从P2到P3，从P3到P0，……。豆豆出发的时候所有食人鱼都在自己路线上的P0位置，请放心，这个位置不会是Start石墩。

Output
输出路线的种数，因为这个数可能很大，你只要输出该数除以10000的余数就行了。 【约定】 1 ≤ N ≤ 50  1 ≤ K ≤ 2,000,000,000  1 ≤ NFish ≤ 20

Sample Input
6 8 1 5 3
0 2
2 1
1 0
0 5
5 1
1 4
4 3
3 5
1
3 0 5 1

Sample Output
2
【样例说明】

时刻 0 1 2 3

食人鱼位置 0 5 1 0

路线一 1 2 0 5

路线二 1 4 3 5

Solution

这题搞了我一下午（周期选错+预处理搞错）。。。
矩阵乘法求两点路线方案数这个模型不必赘述。
但是它有一个食人鱼的时间限制，我一开始脑抽周期选了4，其实应该选2,3,4的最小公倍数12，因为矩阵快速乘的时候的乘起来的两个矩阵应该是一样的。周期选4的话，前4s和后4s的状态肯定不一样。
周期选好，就预处理出一个矩阵，表示每12s，两点间方案数。
然后再用快速幂的形式乘起来，乘Time/12次，最后再乘Time%12的矩阵。
感想：写矩乘一定要清醒并对拍。
时间复杂度：$n^3*log(k/12)$

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define ll long long
using namespace std;
const int mod=10000;
int n,m,st,ed,Time,fish,q,x,y,z,t;
int a[55][55],f[55][55],p[55][55];
int b[15][55],c[15][55][55];
void quick(int y)
{
    for(int i=1;i<=n;i++) f[i][i]=1;
    while(y>0) 
    {
        if(y%2==1) 
        {
            memset(p,0,sizeof(p));
            for(int i=1;i<=n;i++) 
            for(int j=1;j<=n;j++) 
            for(int k=1;k<=n;k++) 
            p[i][k]=(p[i][k]+f[i][j]*c[12][j][k])%mod;
            for(int i=1;i<=n;i++) 
            for(int j=1;j<=n;j++) f[i][j]=p[i][j];
        }
        y=y/2;
        memset(p,0,sizeof(p));
        for(int i=1;i<=n;i++) 
        for(int j=1;j<=n;j++) 
        for(int k=1;k<=n;k++) 
        p[i][k]=(p[i][k]+c[12][i][j]*c[12][j][k])%mod;
        for(int i=1;i<=n;i++) 
        for(int j=1;j<=n;j++) c[12][i][j]=p[i][j];
    }
}         
int main()
{
    cin>>n>>m>>st>>ed>>Time;
    st++;ed++;
    for(int i=1;i<=m;i++) 
    {
        scanf("%d%d",&x,&y);
        x++;y++;
        a[x][y]=1;
        a[y][x]=1;
    }
    cin>>fish;
    for(int i=1;i<=fish;i++) 
    {
        scanf("%d",&q);
        if(q==2) 
        {
            scanf("%d%d",&x,&y);
            x++;y++;
            b[1][y]=1;b[2][x]=1;b[3][y]=1;b[4][x]=1; //b[i][j]表示i秒时j不能走
            b[5][y]=1;b[6][x]=1;b[7][y]=1;b[8][x]=1;
            b[9][y]=1;b[10][x]=1;b[11][y]=1;b[12][x]=1;
        }
        if(q==3) 
        {
            scanf("%d%d%d",&x,&y,&z);
            x++;y++;z++;
            b[1][y]=1;b[2][z]=1;b[3][x]=1;b[4][y]=1;
            b[5][z]=1;b[6][x]=1;b[7][y]=1;b[8][z]=1;
            b[9][x]=1;b[10][y]=1;b[11][z]=1;b[12][x]=1;
        }
        if(q==4) 
        {
            scanf("%d%d%d%d",&x,&y,&z,&t);
            x++;y++;z++;t++;
            b[1][y]=1;b[2][z]=1;b[3][t]=1;b[4][x]=1;
            b[5][y]=1;b[6][z]=1;b[7][t]=1;b[8][x]=1;
            b[9][y]=1;b[10][z]=1;b[11][t]=1;b[12][x]=1;
        }
    }
    for(int i=1;i<=n;i++) 
    for(int j=1;j<=n;j++) 
    if(b[1][j]==0) c[1][i][j]=a[i][j];
    for(int w=2;w<=12;w++) //预处理矩阵
    for(int i=1;i<=n;i++)  
    for(int j=1;j<=n;j++) 
    for(int k=1;k<=n;k++) 
    if(b[w][k]==0) c[w][i][k]=(c[w][i][k]+c[w-1][i][j]*a[j][k])%mod;
    quick(Time/12); //快速矩阵乘法
    if(Time%12>0) 
    {
        memset(p,0,sizeof(p));
        for(int i=1;i<=n;i++) 
        for(int j=1;j<=n;j++) 
        for(int k=1;k<=n;k++) 
        p[i][k]=(p[i][k]+f[i][j]*c[Time%12][j][k])%mod;
        for(int i=1;i<=n;i++) 
        for(int j=1;j<=n;j++) f[i][j]=p[i][j];
    }
    cout<<f[st][ed];
    return 0;
}