ybtoj 矩阵乘法 行为方案

ybtoj 矩阵乘法 行为方案

题面描述

我们可以从题目中得到一些类似于dp的一些好东西。
对于题目中的三种要求我们可以进行如下转化，对于自爆这一种情况我们可以看作是走到了第零号城市，也就是零号节点，如果对与站在原地不动，那么我们就可以默认为它走了一次自环，，那么如果设f[i][j][k]为从i节点走到j节点用了k步数的方案数，那么我们可以得到一条状态转移方程

$f[i][j][k]=\sum f[i][p][k-1]\ast f[p][j][1]$

通过一定的观察可以得到，上面的k-1和1是可以省略掉的，也就是说此时就变成了二维
那么柿子就变成了这样。

$f[i][j]=\sum f[i][p]\ast f[p][j]$

再次观察可以得到这与矩阵乘法的运算法则好像很相似（其实就是），那么我们就可以愉快的用矩阵乘法来解决问题，那么答案就在最后矩阵快速幂后1与各个节点之和

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const ll M=1e2+11,mod=2017;
ll n,m,t;
ll u,v,g;
struct matrix
{
	ll num[M][M];
	ll xl,yl;
}mat;
matrix operator *(const matrix &a,const matrix &b)
{
	matrix c;
	memset(c.num,0,sizeof(c.num));
	c.xl=a.xl,c.yl=b.yl;
	for(int k=0;k<=a.yl;++k)
	for(int i=0;i<=a.xl;++i)
	for(int j=0;j<=b.yl;++j)
		c.num[i][j]=(c.num[i][j]+(a.num[i][k]*b.num[k][j])%mod)%mod;
	return c;
}
matrix unit(matrix f)
{
	f.xl=f.yl=n;
	memset(f.num,0,sizeof(f.num));
	for(int i=0;i<=n;++i) f.num[i][i]=1;
	return f;
}
matrix ksm(matrix x,ll y)
{
	matrix res;
	res=unit(res);
	while(y)
	{
		if(y&1) res=res*x;
		x=x*x,y>>=1;
	}
	return res;
}
int main()
{
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=m;++i)
	{
		scanf("%lld%lld",&u,&v);
		mat.num[u][v]=1,mat.num[v][u]=1;
	}
	for(int i=0;i<=m;++i) mat.num[i][i]=mat.num[i][0]=1;//自环和爆炸节点
	//默认走到零号节点为自爆
	mat.xl=mat.yl=n;
	scanf("%lld",&t);
	matrix ans=ksm(mat,t);
	for(int i=0;i<=n;++i) g=(g+ans.num[1][i])%mod;
	printf("%lld",g);
	return 0;
}