[TJOI2017] 可乐

bzoj 4887 传送门

水题啊......

把原地不动看成是自环，把自爆看作是走到了0点就行了。

所有点都能走到0点，而0点只能走到自己。

然后邻接矩阵自乘即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int mod=2017;

struct matrix
{
    int a[35][35];
}e;

int n,m,t;

matrix mul(matrix &q,matrix &w)
{
    matrix ret;
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=n;j++)
        {
            ret.a[i][j]=0;
            for(int k=0;k<=n;k++)
                ret.a[i][j]+=q.a[i][k]*w.a[k][j];
            ret.a[i][j]%=mod;
        }
    }
    return ret;
}

matrix ksm(matrix b,int p)
{
    matrix ret=b;
    p--;
    while(p)
    {
        if(p&1)ret=mul(ret,b);
        b=mul(b,b);
        p>>=1;
    }
    return ret;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        e.a[x][y]=e.a[y][x]=1;
    }
    for(int i=1;i<=n;i++)
    {
        e.a[i][0]=e.a[i][i]=1;
    }
    e.a[0][0]=1;
    scanf("%d",&t);
    matrix ans=ksm(e,t);
    int sum=0;
    for(int i=0;i<=n;i++)
        sum=(sum+ans.a[1][i])%mod;
    printf("%d",sum);
    return 0;
}