luogu P1357 花园

题目传送门：https://www.luogu.org/problemnew/show/P1357#sub

题意：

有n个花圃，任意相邻m个花圃中有不超过k个C形的花圃，其余花圃均为P形的花圃。求可行的方案数。

思路：

我们看到题目好像很难。

全排列+矩阵乘法。

我们想到，既然要满足题意，我们就可以弄一个1~m的全排列来找出在n个花圃中C、P的排列方案。然后用矩阵乘法搞搞就可以了。

为什么呢？

设i到k1有x1种方案，k1到j有x2种方案，那么i到j有(x1*x2)种方案；

设j到k2有y1种方案，k2到j有y2种方案，那么i到j有(y1*y2)种方案。

所以i到j有x1*x2+y1*y2种方案。

因为又有n个，而我们只算了m个，所以用矩阵乘法呀。

代码：

#include<cstdio>
#include<cstring> 
#define LL long long
#define mod 1000000007
	LL n,m,k,p,sum=0;
	bool bz[100];
	LL a[100][100],ans[100][100];
void work(int y,int t)//表示已经选了y个C,t表示当前状态(二进制数,每一位的0表示选了P,1表示选了C)
{
	bz[t]=true;
	int o=(t>>1);//丢掉最右边的一位,在最左边加上一个0,相当于状态转移 
	a[o][t]=1;//o到t有1种方案 
	if(y==k&&!(t&1)) return;//如果不满足"任意相邻m个花圃中有不超过k个C形的花圃"就return 
	a[o|(1<<m-1)][t]=1;//同理 
}
void dfs(int x,int y,int t)//x表示当前已经选了x个,表示已经选了y个C,t表示当前状态(二进制数,每一位的0表示选了P,1表示选了C)
{
	if(x==m)
	{
		work(y,t);
		return;
	}
	dfs(x+1,y,t);//在x位选P花圃 
	if(y<k) dfs(x+1,y+1,t|(1<<x));//在第x位选C花圃
}
void cheng(LL c[100][100],LL b[100][100])
{
	LL p[100][100];
	memset(p,0,sizeof(p));
	for(int i=0;i<1<<m;i++)
		for(int j=0;j<1<<m;j++)
			for(int k=0;k<1<<m;k++)
				p[i][j]=(p[i][j]+c[i][k]*b[k][j]%mod)%mod;
	for(int i=0;i<1<<m;i++)
		for(int j=0;j<1<<m;j++)
			c[i][j]=p[i][j];
}
int main()
{
	scanf("%lld %lld %lld",&n,&m,&k);
	dfs(0,0,0);
	bool flag=true;
	while(n)
	{
		if(flag)
		{
			for(int i=0;i<1<<m;i++)
				for(int j=0;j<1<<m;j++)
					ans[i][j]=a[i][j];
			flag=false;
			n--;
			continue;
		}
		if(n&1) cheng(ans,a);
		cheng(a,a);
		n>>=1;	
	}
	for(int i=0;i<1<<m;i++)
		if(bz[i]) sum=(sum+ans[i][i])%mod;
	printf("%lld",sum);
} 

小L有一座环形花园，沿花园的顺时针方向，他把各个花圃编号为1~N(2<=N<=10^15)。他的环形花园每天都会换一个新花样，但他的花园都不外乎一个规则，任意相邻M(2<=M<=5,M<=N)个花圃中有不超过K(1<=K<M)个C形的花圃，其余花圃均为P形的花圃。