hdu6304 2018杭电多校第二场J题 Matrix

hdu6304 2018杭电多校第二场J题 Matrix

http://acm.hdu.edu.cn/showproblem.php?pid=6314

题意：

这一题的题意很简单，就是问你，给一个$n\times m$的矩阵涂色，每一格只能涂成黑或者白。
问你，至少有A行和B列全黑的涂色方法有多少种？

思路：

现在不妨假设有$u$行和$x$列全黑，这个有多少种涂色方法呢？
我们可以这样想:我们把全黑的行和列分别压到矩阵的下面和右边，那么除去全黑的行和列就会出现一个新的矩阵，并且这个矩阵满足任意的行和列都不能全黑。
那么这个新矩阵的涂色方法又有多少种呢？我们可以容斥一下，用所有可能的情况减去至少有一行或一列全黑的情况，那么这个新矩阵的涂色方法就有：
${\sum }_{i=0}^{n-u}{\sum }_{j=0}^{m-x}(-1{)}^{i+j}{C}_{n-u}^i{C}_{m-x}^j{2}^{(n-u-i)(m-x-j)}$
然后有$u$行和$x$列全黑时的情况就有：
$C_n^u\ast C_m^x\ast ({\sum }_{i=0}^{n-u}{\sum }_{j=0}^{m-x}(-1{)}^{i+j}{C}_{n-u}^i{C}_{m-x}^j{2}^{(n-u-i)(m-x-j)})$
那么对于至少A行和B列全黑的涂色方法数就可以写成这样：
${\sum }_{u=a}^n{\sum }_{x=b}^m{C}_n^u\ast C_m^x\ast ({\sum }_{v=0}^{n-u}{\sum }_{y=0}^{m-x}(-1{)}^{v+y}{C}_{n-u}^v{C}_{m-x}^y{2}^{(n-u-v)(m-x-y)})$

$\Rightarrow {\sum }_{u=a}^n{\sum }_{x=b}^m{\sum }_{v=0}^{n-u}{\sum }_{y=0}^{m-x}{C}_n^u\ast C_m^x\ast (-1{)}^{v+y}\ast C_{n-u}^v\ast C_{m-x}^y{2}^{(n-u-v)(m-x-y)}$

$\Rightarrow {\sum }_{u=a}^n{\sum }_{x=b}^m{\sum }_{v=0}^{n-u}{\sum }_{y=0}^{m-x}\frac{n!m!(-1{)}^{v+y}{2}^{(n-u-v)(m-x-y)}}{u!v!(n-u-v)!x!y!(m-x-y)!}$

$\Rightarrow n!m!({\sum }_{u=a}^n{\sum }_{v=0}^{n-u}\frac{(-1{)}^v}{u!v!w!}\ast 2^w)({\sum }_{x=b}^m{\sum }_{y=0}^{m-x}\frac{(-1{)}^y}{x!y!z!}\ast 2^z)(w,z分别为n-u-v,m-x-y)$
知道上面这个式子之后我们可以发现
$u+v=n-w,u\ge a$
$x+y=m-z,x\ge b$
然后我们就可以枚举w,z再预处理这样的一个数组：
pre[s][u]表示u+v=s并且$u\ge a$它的$\frac{(-1{)}^v}{u!v!}$的和
这样式子就相当于
$n!m!{\sum }_{w=0}^{n-a}{\sum }_{z=0}^{m-b}\frac{pre[n-w][a]\ast pre[m-z][b]\ast 2^{wz}}{w!v!}$
然后再把二的指数和阶乘的乘法逆元预处理一下时间就那刚好过了。

#include
#include
using namespace std;
const int maxn=3e3+5;
const long long mod=998244353;
int bit[9000006];
long long jc[maxn];
int cal[maxn][maxn];
long long inv[maxn];
long long pre[maxn][maxn];
int n,m;
long long qpow(long long a,long long b)
{
	long long ans=1;
	while(b>0)
	{
		if(b&1) ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}
void init()
{
	int i,j;
	int val;
	bit[0]=1;
	jc[0]=1;
	inv[0]=1;
	for(i=1;i<=9000000;i++)
		bit[i]=bit[i-1]*2%mod;
	for(i=1;i<=3001;i++)
	{
		jc[i]=jc[i-1]*i%mod;
		inv[i]=qpow(jc[i],mod-2);
		// cout<
	}
	for(i=0;i<=3000;i++)
		for(j=0;j<=3000;j++)
			cal[i][j]=bit[i*j]*inv[i]%mod*inv[j]%mod;
	for(i=0;i<=3001;i++)
	{
	 	for(j=i;j>=0;j--)
		{
			val=inv[j]*inv[i-j]%mod;
			if((i-j)%2) val = mod-val;
	 		pre[i][j]=(pre[i][j+1]+val)%mod;
		}
	}
}

int main()
{
	int i,j,k,n,m,a,b;
	long long ans=0;
	init();
	while(scanf("%d%d%d%d",&n,&m,&a,&b)!=EOF)
	{
		ans=0;
		for(int w=0;w<=n-a;w++)
		{
			for(int z=0;z<=m-b;z++)
			{
				// cout<
				ans=(ans+pre[n-w][a]*pre[m-z][b]%mod*cal[w][z])%mod;
			}
		}
		// printf("%lld\n",ans );
		printf("%lld\n",ans*jc[n]%mod*jc[m]%mod);
	}
}