P6076-[JSOI2015]染色问题【组合数学,容斥】

1|0正题

题目链接:https://www.luogu.com.cn/problem/P6076

---

1|1题目大意

给出$n*m$的网格，$c$种颜色涂色要求

1. 每个格子可以染色也可以不染
2. 每一行每一列至少有一个格子被染
3. 每个颜色至少用一次

$1\leq n,m,c\leq 400$

---

1|2解题思路

一个比较简单的方法就是容斥，枚举有多少染色的和不染色的行列，和枚举使用的颜色个数

$$\sum_{i=0}^c\sum_{j=0}^n\sum_{k=0}^m\binom{c}{i}\binom nj\binom mk(i+1)^{j+k}(-1)^{c+n+m-i-j-k}$$

这样预处理就是$O(nmc)$的，但是可以做到更快。

设$f_i$表示最多染了$i$种颜色的方案，那么久只需要满足第二个条件了。第二个条件可以用一个容斥搞定，考虑枚举多少行没染

$$f_k=\sum_{i=1}^n(-1)^{n-i}((k+1)^{m}-1)^i$$

这样预处理就可以做到$O(nc)$

这里写的是第一种，因为比较懒

---

1|3code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=410,P=1e9+7;
ll n,m,c,C[N][N],pw[N*N],ans;
signed main()
{
	scanf("%lld%lld%lld",&n,&m,&c);
	C[0][0]=1;
	for(ll i=1;i<N;i++)
		for(ll j=0;j<N;j++)
			C[i][j]=(C[i-1][j]+(j?C[i-1][j-1]:0))%P; 
	for(ll i=0;i<=c;i++){
		pw[0]=1;
		for(ll j=1;j<=n*m;j++)
			pw[j]=pw[j-1]*(i+1)%P;
		for(ll j=0;j<=n;j++)
			for(ll k=0;k<=m;k++){
				ll f=(c-i)+(n-j)+(m-k);
				if(f&1)f=-1;else f=1;
				(ans+=f*C[n][j]*C[m][k]%P*C[c][i]%P*pw[j*k]%P)%=P;
			}
	}
	printf("%lld\n",(ans+P)%P);
	return 0;
}

__EOF__

本文作者：QuantAsk
本文链接：https://www.cnblogs.com/QuantAsk/p/14550240.html
关于博主：退役OIer，GD划水选手
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！