jzoj3206 [HNOI模拟题] Axonometric Projection 容斥

Description

---

给出一堆小单位立方体的正视图和侧视图，求合法摆放方案数

Solution

---

可以发现我们交换限制不会影响答案

考虑把限制分别排序，那么相同高度的位置一定形成了若干L字形。由于限制是取max的，因此这些L字形之间互不影响。

我们单独算某一个高度为i的L字形，枚举至少i行j列不合法，那么分两部分算
上图中，BC两部分是限制相等的位置，C是我们枚举的不合法的位置

1. C中我们只能选[0,h-1]，那么就是h^(C的面积)
2. B中我们随便选，那么就是h^(B的面积)

猜一波可以发现容斥系数恰好就是${\left(-1\right)}^{i+j}$，由于限制都是等价的因此我们还要乘上组合数

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)

typedef long long LL;
const int MOD=1000000009;
const int N=20005;

LL a[N],b[N],fac[N],ny[N];

LL ksm(LL x,LL dep) {
	LL res=1;
	for (;dep;dep>>=1) {
		(dep&1)?(res=res*x%MOD):0;
		x=x*x%MOD;
	}
	return res;
}

LL C(int n,int m) {
	return fac[n]*ny[m]%MOD*ny[n-m]%MOD;
}

LL calc(LL n,LL m,LL p,LL q,LL h) {
	LL res=0;
	rep(i,0,p) {
		rep(j,0,q) {
			LL tmp=C(p,i)*C(q,j)%MOD;
			tmp=tmp*ksm(h,n*m-(n-i)*(m-j))%MOD;
			tmp=tmp*ksm(h+1,(n-i)*(m-j)-(n-p)*(m-q))%MOD;
			if ((i+j)&1) tmp=((MOD-tmp)%MOD+MOD)%MOD;
			res=(res+tmp)%MOD;
		}
	}
	return res;
}

int main(void) {
	fac[0]=1; rep(i,1,N-1) fac[i]=1LL*i*fac[i-1]%MOD;
	ny[N-1]=ksm(fac[N-1],MOD-2);
	drp(i,N-2,0) ny[i]=1LL*(i+1)*ny[i+1]%MOD;
	int n; scanf("%d",&n);
	rep(i,1,n) {
		int x; scanf("%d",&x);
		a[x]++;
	}
	int m; scanf("%d",&m);
	rep(i,1,m) {
		int x; scanf("%d",&x);
		b[x]++;
	}
	LL sa=0,sb=0,ans=1;
	drp(i,10000,1) {
		if (a[i]||b[i]) {
			sa+=a[i],sb+=b[i];
			ans=ans*calc(sa,sb,a[i],b[i],i)%MOD;
		}
	}
	printf("%lld\n", ans);
	return 0;
}