bzoj 4487: [Jsoi2015]染色问题

最新推荐文章于 2023-06-18 22:35:08 发布

fyc_kabuto

最新推荐文章于 2023-06-18 22:35:08 发布

阅读量356

点赞数 1

分类专栏：容斥原理

本文链接：https://blog.csdn.net/qq_36808030/article/details/79097979

版权

容斥原理专栏收录该内容

27 篇文章 0 订阅

订阅专栏

题意：

问在 $n*m$ 的网格图上染色，要求每一行，每一列至少有一个格被染色，且每种颜色至少出现一次，问方案数。

题解：

今天脑子非常不清醒，想这种容斥裸题都想了半天……一开始还写了个 $n^3$ 的。
直接做不好做，显然要容斥，有三个限制，其中两个容斥，剩下一个算时候保证合法。
我先对列容斥，设 $f[i]$ 表示最多有 $i$ 列有格子被染色过
那么严格 $i$ 列就是 $f'[i]=\sum_{j=1}^i(-1)^{i-j}f[j]*C_i^j$
考虑如何求 f <script type="math/tex" id="MathJax-Element-502">f</script>，因为每种颜色至少要有一次，所以再对颜色用相同的方法容斥就可以了。
code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
const LL mod=1000000007;
using namespace std;
LL C[410][410],n,m,c,f[410],g[410];
void pre()
{
    for(LL i=0;i<=400;i++)
    {
        C[i][0]=1;
        for(LL j=1;j<=i;j++) C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
    }
}
LL pow(LL a,LL b)
{
    LL ans=1;
    while(b) {if(b&1) ans=ans*a%mod;a=a*a%mod;b>>=1;}
    return ans;
}
LL solve(LL M)
{
    for(LL i=1;i<=c;i++) g[i]=pow(pow(i+1,M)-1,n);
    LL ans=0;
    for(int j=c;j>=1;j--) ans=(ans+(((c-j+1)&1)?1:-1)*g[j]*C[c][j]%mod)%mod,(ans+=mod)%=mod;
    return ans;
}
int main()
{
    scanf("%lld %lld %lld",&n,&m,&c);pre();
    for(LL i=1;i<=m;i++) f[i]=solve(i);
    LL ans=0;
    for(int j=m;j>=1;j--) ans=(ans+(((m-j+1)&1)?1:-1)*f[j]*C[m][j]%mod)%mod,(ans+=mod)%=mod;
    printf("%lld",ans);
}