【ybtoj 高效进阶 6.1】 【矩乘】 行为方案 【洛谷 P3758】 可乐

【ybtoj 高效进阶 6.1】 【矩乘】 行为方案 【洛谷 P3758】 可乐

题目

---

解题思路

首先可以想到设f[i][j][k]为从第i个城市走到第j个城市走了k步的方案数
因为f[i][j][k]只会和f[i][q][k]有关系，那么就可以优化掉第三维
得

长得就很像矩乘啊
用矩乘求出来这个方程处理t次
最后答案为1到各个节点的方案数

---

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int mo=2017;
struct lzf{
	int n,m,l[112][112];
}a,d;
int n,m,x,y,s,ans; 
lzf operator * (lzf a,lzf b) //重新定义乘号
{
	lzf c;
	c.n=a.n,c.m=b.m;
	memset(c.l,0,sizeof(c.l));
	for (int i=0;i<=c.n;i++)
	    for (int j=0;j<=c.m;j++)
	        for (int k=0;k<=a.m;k++)
	            c.l[i][j]=(c.l[i][j]+a.l[i][k]*b.l[k][j])%mo;
	return c;
}
void power(int x)  //快速幂
{
	 d.n=n,d.m=n;
	 for (int i=0;i<=n;i++) d.l[i][i]=1;
	 while (x)
	 {
	 	   if (x&1) d=d*a;
	 	   a=a*a;
	 	   x=x/2;
	 }
} 
int main()  
{
	scanf("%d%d",&n,&m);
	a.n=n,a.m=n;
	for (int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		a.l[x][y]=a.l[y][x]=1;
	}
	for (int i=0;i<=n;i++)
		a.l[i][i]=a.l[i][0]=1;  //自爆的情况
	scanf("%d",&s);
	power(s);
	for (int i=0;i<=n;i++)
	    ans=(ans+d.l[1][i])%mo;
	printf("%d\n",ans);
	return 0;
}