Luogu

分析

由熟知结论,邻接矩阵的 $k$ 次幂中 $(u, v)$ 上的数表示 $u\to v$ 恰好走 $k$ 步的方案数。

然而本题有一个不动和自爆的问题。

不动的话,相当于走自环;自爆的话,减一个 $0$ 号节点,$0 \sim n$ 都向 $0$ 连一条边即可。这样子保证了自爆后就没有后继了。

代码

#include <bits/stdc++.h>
#define re register
using namespace std;

inline int read() {
    int X=0,w=1; char c=getchar();
    while (c<'0'||c>'9') { if (c=='-') w=-1; c=getchar(); }
    while (c>='0'&&c<='9') X=(X<<3)+(X<<1)+c-'0',c=getchar();
    return X*w;
}

const int MOD=2017;

struct Matrix {
    int m[50][50];
};

Matrix Mul(Matrix a,Matrix b,int n) {
    Matrix c;
    for (re int i=0;i<=n;i++)
        for (re int j=0;j<=n;j++) {
            c.m[i][j]=0;
            for (re int k=0;k<=n;k++)
                c.m[i][j]=(c.m[i][j]+a.m[i][k]*b.m[k][j])%MOD;
        }
    return c;
}

Matrix FastPow(Matrix a,int k,int n) {
    Matrix rt;
    for (re int i=0;i<=n;i++)
        for (re int j=0;j<=n;j++) {
            if (i==j) rt.m[i][j]=1;
            else rt.m[i][j]=0;
        }
    while (k) {
        if (k&1) rt=Mul(rt,a,n);
        a=Mul(a,a,n);
        k>>=1;
    }
    return rt;
}

Matrix G;

int main() {
    int n=read(),m=read();
    for (re int i=1;i<=m;i++) {
        int a=read(),b=read();
        G.m[a][b]=G.m[b][a]=1;
    }
    for (re int i=0;i<=n;i++) G.m[i][i]=1;
    for (re int i=1;i<=n;i++) G.m[i][0]=1;
    int t=read();
    G=FastPow(G,t,n);
    int ans=0;
    for (re int i=0;i<=n;i++) ans=(ans+G.m[1][i])%MOD;
    printf("%d\n",ans);
    return 0;
}
最后修改:2021 年 03 月 23 日 08 : 56 AM