【BZOJ4498】魔法的碰撞（DP）（组合数学）

传送门

---

题解：

首先将$d_i$全部$-1$，方便下面叙述。

首先我们假设某个排列$p$中，魔法师之间站得尽可能紧之后中间空的格子个数为$x=\sum _{i=1}^{n-1}\max (d_{p_i},d_{p_{i+1}})$（显然这时候格子是最少的），那么按照这个顺序的来所有方案数为$\left(\genfrac{}{}{0ex}{}{L-x}{n}\right)$，我们只需要在剩下的自由格子里面随便选择就行了。

于是可以暴力枚举所有排列，发现距离的可能的值远远比总的排列的值要小，考虑算出$cnt[x]$，即最终空了$x$格的排列数。

我们考虑将$d$排序，从大到小插入，发现小的对大的贴脸插入的时候能够不产生任何贡献，我们需要考虑小的可能从中间插入，需要预留空位。

令$f[i][j][k]$表示考虑了前$i$个，预留了$j$个空位，空格有$k$个的排列数，转移只需要考虑当前这个点插入某个空位之后是否在两端留下空位就行了。

复杂度是$O({\sum }_{i=1}^n{d}_{i}r{k}_i)$，其中$r{k}_i$表示第$i$个$d$值是第几大。

总的来说，复杂度不会超过$O(n^2\sum d)$

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

cs int mod=1e9+7;
inline int add(int a,int b){return (a+=b)>=mod?a-mod:a;}
inline int dec(int a,int b){return (a-=b)<0?a+mod:a;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){(a+=b)>=mod&&(a-=mod);}
inline void Dec(int &a,int b){(a-=b)<0&&(a+=mod);}

cs int N=42,P=1e6+6;
int n,L;
int d[N];
int f[N][N][N*N*2+10];
int fac[P],ifac[P];

inline int C(int n,int m){return n>=0&&m>=0&&n>=m?mul(fac[n],mul(ifac[m],ifac[n-m])):0;}

signed main(){
//	freopen("magic.in","r",stdin);
	scanf("%d%d",&L,&n);
	for(int re i=1;i<=n;++i)scanf("%d",d+i),d[i]--;
	fac[0]=ifac[0]=1;
	for(int re i=1;i<=L;++i)fac[i]=mul(fac[i-1],i);
	ifac[L]=power(fac[L],mod-2);
	for(int re i=L-1;i;--i)ifac[i]=mul(ifac[i+1],i+1);
	
	std::sort(d+1,d+n+1);std::reverse(d+1,d+n+1);
	f[0][1][0]=1;int s=0;
	for(int re i=1;i<=n;++i){
		s+=d[i];
		for(int re j=0;j<=i+1;++j)
		for(int re k=0;k<=2*s;++k){
			f[i][j][k]=mul(j+1,f[i-1][j+1][k]);
			if(k>=d[i])Inc(f[i][j][k],mul(j<<1,f[i-1][j][k-d[i]]));
			if(j&&k>=2*d[i])Inc(f[i][j][k],mul(j-1,f[i-1][j-1][k-d[i]*2]));
		}
	}
	int ans=0;
	for(int re i=0,lim=std::min(2*s,L-n);i<=lim;++i)Inc(ans,mul(f[n][0][i],C(L-i,n)));
	cout<<ans<<"\n";
	return 0;
}