【BZOJ4887】[TJOI2017]可乐（矩阵快速幂）

【BZOJ4887】[TJOI2017]可乐（矩阵快速幂）

题面

BZOJ
洛谷

题解

模板题？？？

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAX 35
#define MOD 2017
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
int g[MAX][MAX],n,m,N;
struct Matrix{int s[MAX][MAX];int*operator[](int x){return s[x];}}A,S;
Matrix operator*(Matrix a,Matrix b)
{
	Matrix c;memset(c.s,0,sizeof(c.s));
	for(int i=1;i<=N;++i)
		for(int j=1;j<=N;++j)
			for(int k=1;k<=N;++k)
				c[i][j]=(c[i][j]+a[i][k]*b[k][j])%MOD;
	return c;
}
int main()
{
	n=read();m=read();N=n+1;
	for(int i=1,u,v;i<=m;++i)u=read(),v=read(),A[u][v]+=1,A[v][u]+=1;
	for(int i=1;i<=n;++i)A[i][i]+=1,A[i][N]+=1;A[N][N]+=1;
	for(int i=1;i<=N;++i)S[i][i]=1;
	int T=read();while(T){if(T&1)S=S*A;A=A*A;T>>=1;}
	int ans=0;for(int i=1;i<=N;++i)ans=(ans+S[1][i])%MOD;
	printf("%d\n",ans);
	return 0;
}