AT2000 Leftmost Ball dp+组合数学

原题:https://www.luogu.org/problemnew/show/AT2000

题解:其实就是方n个白球和k-1个其他n种颜色的求的方案数，可以发现不需要关注每个球的位置，考虑转化f[i][j]表示放了i个白球，j种颜色的球(i>=j) 考虑左边数第一个空位放什么，放白球时f[i][j]+=f[i-1][j]，放其他颜色时应该从放f[i][j-1]共有n-(j-1)个颜色其他球，有n*k-i-(j-1)*(k-1)-1个空位，根据乘法原理f[i][j]+=(n-j+1)*C(n*k-i-(j-1)*(k-1)-1)*f[i][j-1]。dp的初值为f[0][0]=1。注意k==1时的特判。

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int N=2200;
const int MAXN=4e6;
const int M=1e9+7;
ll fac[N*N],inv[N*N],f[N][N];
int n,k;
inline ll qpow(ll a,ll b){
	ll ans=1;
	while(b){
		if(b&1) ans=ans*a%M;
		b>>=1;a=a*a%M;
	} 
	return ans;
}
void init(){
	fac[0]=1;
	for(int i=1;i<=MAXN;i++) fac[i]=fac[i-1]*i%M;
	inv[MAXN]=qpow(fac[MAXN],M-2); 
	for(int i=MAXN;i>=1;i--) inv[i-1]=inv[i]*i%M;
}
inline ll C(int n,int m){
	if(n<m) return 0;
	return fac[n]*inv[m]%M*inv[n-m]%M;
}
int main(){
	init();
	scanf("%d%d",&n,&k);f[0][0]=1;
	if(k==1) {printf("1\n");return 0; } 
	for(int i=1;i<=n;i++)
		for(int j=0;j<=i;j++) f[i][j]=(f[i-1][j]+(n-j+1)*C(n*k-i-(j-1)*(k-1)-1,k-2)%M*f[i][j-1])%M;
	printf("%lld\n",f[n][n]);
	return 0;
}