【BZOJ 3622】已经没有什么好害怕的了

题目链接

题意简述

给你两个长度为n的数组A和B,其中数字各不相同,要求将数字之间一一匹配,要求 A 比 B 大的对数比 B 比 A 大的对数多 K 对,求方案数

Sol

难得自己写出计数题,这题还是比较简单的

显然可以把问题转化为选出 $(n+k)/2$ 对 A 比 B 大的对
先考虑 dp, 我们可以先排一个序,这样就能快速知道当前有多少个数比自己小,$f[i][j]$ 表示A 数组到了$i$ , 已经有 $j$ 个对是 A 大的了,YY转移:
$f[i][j+1]=f[i-1][j]\ast (cnt-j)cnt$表示当前有 $cnt$ 个数比自己小
$f[i][j]=f[i-1][j]\ast (n-...)$

选择比自己大的匹配的情况不好考虑,因为可能之前比原来的数大的现在比当前数小 , 那么就不好算了 , 不过考虑到我们最后一定是剩下 $n-k$ 对要选择比自己大的匹配 , 那么最后乘上一个 $(n-k)!$ 去匹配即可

显然直接这样做是有问题的,我们不能保证最后不产生又选择了比自己小的数的情况 , 那么这是我们的状态表示的其实是至少选了 $j$ 对比自己小的匹配的情况 , 二项式反演一下就好了。

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
template<class T>inline void init(T&x){
	x=0;char ch=getchar();bool t=0;
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') t=1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=(x<<1)+(x<<3)+(ch-48);
	if(t) x=-x;return;
}
const int N=2001;
const int mod=1e9+9;
template<class T>inline void Inc(T&x,int y){x+=y;if(x>=mod) x-=mod;}
template<class T>inline void Dec(T&x,int y){x-=y;if(x < 0 ) x+=mod;}
int n,k;
int A[N],B[N],f[N][N],inv[N],fac[N];
int main()
{
	init(n),init(k);inv[0]=inv[1]=1;
	k=n+k;if(k&1) return puts("0"),0;k>>=1;
	for(int i=1;i<=n;++i) init(A[i]);
	for(int i=1;i<=n;++i) init(B[i]);
	fac[0]=fac[1]=1;
	for(int i=2;i<=n;++i) inv[i]=(ll)(mod-mod/i)*inv[mod%i]%mod,fac[i]=(ll)fac[i-1]*i%mod;
	sort(A+1,A+1+n);sort(B+1,B+1+n);
	int p=1;
	f[0][0]=1;
	for(int i=1;i<=n;++i) {
		while(B[p]<A[i]&&p<=n) ++p;int num=p-1;
		for(int j=0;j<=n;++j) {
			if(num>j) Inc(f[i][j+1],(ll)f[i-1][j]*(num-j)%mod);
			Inc(f[i][j],f[i-1][j]);
		}
	}
	int ans=(ll)f[n][k]*fac[n-k]%mod;
	int C=1;
	for(int i=k+1;i<=n;++i) {
		C=(ll)C*i%mod*inv[i-k]%mod;
		((i-k)&1)? Dec(ans,(ll)C*f[n][i]%mod*fac[n-i]%mod):Inc(ans,(ll)C*f[n][i]%mod*fac[n-i]%mod);
	}
	cout<<ans<<endl;
}