bzoj4487[Jsoi2015]染色问题 容斥+组合

4487: [Jsoi2015]染色问题

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 211  Solved: 127
[Submit][Status][Discuss]

Description

棋盘是一个n×m的矩形，分成n行m列共n*m个小方格。现在萌萌和南南有C种不同颜色的颜料，他们希望把棋盘用这些颜料染色，并满足以下规定：
1.  棋盘的每一个小方格既可以染色（染成C种颜色中的一种） ，也可以不染色。
2.  棋盘的每一行至少有一个小方格被染色。
3.  棋盘的每一列至少有一个小方格被染色。
4.  种颜色都在棋盘上出现至少一次。
以下是一些将3×3棋盘染成C = 3种颜色（红、黄、蓝）的例子：

请你求出满足要求的不同的染色方案总数。只要存在一个位置的颜色不同，
即认为两个染色方案是不同的

Input

输入只有一行 3 个整数n,m,c。1 < = n,m,c < = 400

Output

输出一个整数，为不同染色方案总数。因为总数可能很大，只需输出总数
mod 1,000,000,007的值。

Sample Input

2 2 3

Sample Output

60

HINT

Source

 

由一维容斥推到三维容斥。。
很诡异,并不是很懂,感性理解
枚举ijk,表示占据i行j列k个颜色或不涂色任意选
容斥就好了。
这样推出式子是O(N^3),根据二项式定理可以优化至O(N^2*log2(M))

看blog
http://blog.csdn.net/nirobc/article/details/51064832

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define N 403
#define p 1000000007
#define LL long long
using namespace std;
int n,m,c;
LL C[N][N];
LL quickpow(int num,int x)
{
LL ans=1,base=num;
while (x) {
if (x&1) ans=ans*base%p;
x>>=1;
base=base*base%p;
}
return ans;
}
int main()
{
scanf("%d%d%d",&n,&m,&c);
for (int i=0;i<=400;i++) C[i][0]=1;
for (int i=1;i<=400;i++)
for (int j=1;j<=i;j++)
C[i][j]=(C[i-1][j]+C[i-1][j-1])%p;
LL ans=0;
for (int k=0;k<=c;k++) {
LL x=1;
for (int i=n;i>=0;i--){
LL tot=1;
for (int j=m;j>=0;j--) {
int t=i+j+k;
LL now=C[n][i]*C[m][j]%p*C[c][k]%p*tot%p;
if (t&1) ans-=now;
else ans+=now;
tot=tot*x%p;
}
x=x*(c-k+1)%p;
ans%=p;
}
}
printf("%lldn",(ans%p+p)%p);
}