题目传送门: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形的花圃。