NOI模拟 长寿花【dp】【组合数学】

传送门

题意：有n层，每层a[i]个位置，m个颜色。每个位置可以选择一个颜色。要求相邻颜色不相同，相邻两层颜色集合不完全相同。求方案数。

发现a[i]最大5000，直接n方预处理。

根据套路设表示第i层用了j个颜色的总方案数。

枚举第i位是否启用新颜色得到

然后再来考虑很多层。

设表示前i层中第i层用了j个颜色的总方案数。

如果不管相邻限制，

然而相邻不能完全相同。那就直接减去就行。因为我本层用j个颜色的选色方案中，只有一种会和上一层所选集合完全相同。

所以

递推。

 

注意：模数任意。不能处理组合数逆元，数据1e6更不能m方递推。发现永远都是从m中选k个，展开成阶乘公式发现每一项只比上一项多*(m-i+1)/i。强行分解质因数递推。

#include<bits/stdc++.h>
using namespace std;
#define in read()
#define int long long
int in{
	int cnt=0;char ch=0;
	while(!isdigit(ch)) ch=getchar();
	while(isdigit(ch)) cnt=cnt*10+ch-48,ch=getchar();
	return cnt;
}
const int N=1000000,M=5003;
int g[M][M],f[2][M],n,m,mod,vis[N+3],prime[N+3],cnt,c[M],a[N+3],maxx,box[N+3];
void Seive(int x,int inv){
	if(x==1)return;
	for(int i=1;i<=cnt&&prime[i]<=x;i++)
		while(x%prime[i]==0)box[i]+=inv,x/=prime[i];
}
void prepare1(){
	for(int i=2;i<=N;i++){
		if(!vis[i]) prime[++cnt]=i;
		for(int j=1;j<=cnt&&prime[j]*i<=N;j++){
			vis[prime[j]*i]=1;
			if(i%prime[j]==0)break;
		}
	}
}
void prepare2(){
	for(int i=1;i<=min(m,maxx);i++){
		c[i]=1;Seive(i,-1);Seive(m-i+1,1);
		for(int j=1;j<=cnt&&prime[j]<=m;j++)
			for(int k=1;k<=box[j];k++)
				c[i]=c[i]*prime[j]%mod;
	}
}
void prepare3(){
	g[0][0]=1;
	for(int i=1;i<=maxx;i++)
		for(int j=1;j<=min(m,maxx);j++)
			g[i][j]=(g[i-1][j-1]*j%mod+(j-1)*g[i-1][j]%mod)%mod;
}
signed main(){//令f[i][0]保存第i层的f和。 
	n=in;m=in;mod=in;
	for(int i=1;i<=n;i++)maxx=max(maxx,a[i]=in);
	prepare1();prepare2();prepare3();f[0][0]=1;
//	for(int i=1;i<=10;i++)cout<<c[i]<<" ";cout<<endl;
//	for(int i=1;i<=10;i++)cout<<prime[i]<<" ";cout<<endl;
//	for(int i=1;i<=10;i++){for(int j=1;j<=i;j++)cout<<g[i][j]<<" ";cout<<endl;
//	}
	for(int i=1;i<=n;i++){
		int x=i&1;f[x][0]=0;
		for(int j=1;j<=min(a[i],m);j++){
			f[x][j]=(f[x^1][0]*g[a[i]][j]%mod*c[j]%mod);
			if(a[i-1]>=j)f[x][j]-=f[x^1][j]*g[a[i]][j];(f[x][j]+=mod)%=mod;
			(f[x][0]+=f[x][j])%=mod;
//			cout<<f[x][j]<<" ";
		}//cout<<f[x][0]<<endl;
	}
	cout<<(f[n&1][0]+mod)%mod;
	return 0;
}