洛谷 P3158 [CQOI2011]放棋子(组合数学/容斥/dp)

题目

bzoj3294

在一个m(m<=30)行n(n<=30)列的棋盘里，放c(c<=10)种彩色的棋子，

使得每个格子最多放一个棋子，且不同颜色的棋子不能在同一行或者同一列。

求合法的方案数，答案对1e9+9取模。

例如，n=m=3，有两个白棋子和一个灰棋子，

下面左边两种方法都是合法的，但右边两种都是非法的。

思路来源

hx073269代码

题解

把棋盘考虑成一个二分图，对于每种颜色单独处理，

考虑为第i种颜色分j行k列，则需要n行里选j行，m列里选k列，

方案不会被重复统计，当且仅当选出的j行k列每一行每一列都被至少一个棋子占据，

所以需要统计i行j列放a[x]个棋子的方案数，

考虑容斥，dp[i][j][k]表示j行k列放a[i]个棋子，

使每一行每一列都被至少一个棋子占据的方案数，

根据dp式的定义，只需依次减去，

满足j行里选j’行，k行里选k’列，即j'<=j，k'<=k，

且(j',k')至少有一维比(j,k)小，的dp[i][j'][k']的方案数

这个容斥，一开始被定义成了i行j列放k个棋子的状态，

使得复杂度为O(30*30*30^2*30*30)，无法通过，

考虑到容斥时，棋子的方案数是直接计算的，并不需要从底层递推，

所以，只有10种类型棋子数，复杂度即为O(10*30*30*30*30)

代码

#include<bits/stdc++.h>
using namespace std;
const int N=31,M=13,up=705,mod=1e9+9;
int n,m,c,a[M],ans,fac[N*N],finv[N*N];
int dp[M][N][N],f[N][N][up],sum[N][N][N][up],num;
int C(int n,int m){
	if(n<0 || m<0 || n<m)return 0;
	return 1ll*fac[n]*finv[m]%mod*finv[n-m]%mod;
}
void init(){
	fac[0]=finv[0]=fac[1]=finv[1]=1;
	for(int i=2;i<N*N;++i){
		fac[i]=1ll*fac[i-1]*i%mod;
		finv[i]=1ll*(mod-mod/i)*finv[mod%i]%mod;
	}
	for(int i=2;i<N*N;++i){
		finv[i]=1ll*finv[i-1]*finv[i]%mod;
	}
}
int dfs(int x,int row,int col){
	if(!x)return 1;
	if(!row || !col)return 0;
	int &ans=dp[x][row][col];
	if(~ans)return ans;
	ans=0;
	for(int i=1;i<=row;++i){
		for(int j=1;j<=col;++j){
			ans=(ans+1ll*C(row,i)*C(col,j)%mod*f[x][i][j]%mod*dfs(x-1,row-i,col-j)%mod)%mod;
		}
	}
	return ans;
} 
int main(){
	init();
	memset(dp,-1,sizeof dp);
	scanf("%d%d%d",&n,&m,&c);
	for(int i=1;i<=c;++i){
		scanf("%d",&a[i]);
	} 
	for(int i=1;i<=c;++i){ 
		for(int j=1;j<=n;++j){
			for(int k=1;k<=m;++k){
				f[i][j][k]=C(j*k,a[i]);
				for(int u=1;u<=j;++u){
					for(int v=1;v<=k;++v){
						if(u==j && v==k)continue;
						f[i][j][k]=(f[i][j][k]+mod-1ll*C(j,u)*C(k,v)%mod*f[i][u][v]%mod)%mod; 
					}
				} 
			}
		}
	}
	printf("%d\n",dfs(c,n,m));
	return 0;
}