【BZOJ3622】已经没什么好害怕的了(dp,容斥原理,二项式反演)

首先设有 x x x a > b a>b a>b,由 x − ( n − x ) = k x-(n-x)=k x(nx)=k 解得 x = n + k 2 x=dfrac{n+k}{2} x=2n+k

先将 a a a b b b 数组排序。

然后设 f ( i , j ) f(i,j) f(i,j) 表示前 i i i a a a 中,选了 j j j a > b a>b a>b

显然有:

f ( i , j ) = f ( i − 1 , j ) + [ l i − ( j − 1 ) ] f ( i − 1 , j − 1 ) f(i,j)=f(i-1,j)+ig[l_i-(j-1)ig]f(i-1,j-1) f(i,j)=f(i1,j)+[li(j1)]f(i1,j1)

其中 l i l_i li 表示 b b b 中比 a i a_i ai 小的有多少个。

g i = ( n − i ) ! f ( n , i ) g_i=(n-i)!f(n,i) gi=(ni)!f(n,i),也就是把 n − i n-i ni 个没有匹配的任意分配,那么就能得到 a > b a>b a>b 的组数 ≥ i geq i i 的方案数(有算重)。

a n s i ans_i ansi 表示 x = i x=i x=i 时的答案。

根据容斥原理,容易发现对于 i > j i>j i>j a n s i ans_i ansi g j g_j gj 中被算了 ( i j ) dbinom{i}{j} (ji) 次。

那么有:(可以用二项式反演推出)

a n s j = ∑ i = j n ( − 1 ) i − j ( i j ) g i ans_j=sumlimits_{i=j}^n(-1)^{i-j}inom{i}{j}g_i ansj=i=jn(1)ij(ji)gi

代码如下:

#include<bits/stdc++.h>

#define N 2010
#define ll long long
#define mod 1000000009

using namespace std;

int n,k,a[N],b[N],l[N];
ll fac[N],inv[N],g[N],f[N][N];

ll poww(ll a,ll b)
{
	ll ans=1;
	while(b)
	{
		if(b&1) ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}

void init()
{
	fac[0]=inv[0]=1;
	for(int i=1;i<=n;i++)
	{
		fac[i]=fac[i-1]*i%mod;
		inv[i]=poww(fac[i],mod-2);
	}
}

ll C(int n,int m)
{
	return fac[n]*inv[m]%mod*inv[n-m]%mod;
}

int main()
{
	scanf("%d%d",&n,&k);
	if((n+k)&1)
	{
		puts("0");
		return 0;
	}
	init();
	k=(n+k)/2;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<=n;i++)
		scanf("%d",&b[i]);
	sort(a+1,a+n+1);
	sort(b+1,b+n+1);
	int now=0;
	for(int i=1;i<=n;i++)
	{
		while(now<n&&b[now+1]<a[i]) now++;
		l[i]=now;
	}
	f[0][0]=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<=l[i];j++)
		{
			f[i][j]=f[i-1][j];
			if(j>0) f[i][j]=(f[i][j]+1ll*(l[i]-(j-1))*f[i-1][j-1])%mod;
		}
	}
	for(int i=0;i<=n;i++)
		g[i]=f[n][i]*fac[n-i]%mod;
	ll ans=0;
	for(int i=k;i<=n;i++)
	{
		int tmp=(((i-k)&1)?-1:1);
		ans=((ans+tmp*C(i,k)*g[i]%mod)%mod+mod)%mod;
	}
	printf("%lld
",ans);
	return 0;
}
原文地址:https://www.cnblogs.com/ez-lcw/p/14448652.html