BZOJ 3622 DP+容斥

Description

Input

Output

Sample Input

4 2

5 35 15 45

40 20 10 30

Sample Output

4
HINT

输入的2*n个数字保证全不相同。

还有输入应该是第二行是糖果，第三行是药片

分析：首先考虑直接算出刚好等于k的方案数显然是不太现实的，于是考虑先算出至少为k的方案数，然后观察了下数据范围n<=2000，咦，好像可以Dp，先把a，b数组排序。接着考虑f[i][j]，i表示前i位，j表示糖大于药至少有j对，f[i][j]表示方案数。然后考虑转移f[i][j]=f[i-1][j]+f[i-1][j-1](nxt[i]-(j-1));nxt[i]表示在b数组中小于a[i]的数的个数。DP完成后我们现在要去除算重的部分，因为我们只算出了至少有k对的方案，于是考虑容斥，∑f[n][i]∗C(i,k)∗(−1)^(i-k)（n-i)!,i>=k，因为配了i对后剩（n-i）对自由配对，方案数为（n-i)!，C[i][k]就是从i个选出k个的方案数，相乘就得到重复部分，然后容斥管用套路+-+-….最后得到答案

# include <iostream>
# include <cstdio>
# include <cstdlib>
# include <cstring>
# include <cmath>
# include <vector>
# include <algorithm>
using namespace std;
typedef long long ll;
inline int read()
{
    register int i=0;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        i=(i<<3)+(i<<1)+ch-'0';
        ch=getchar();
    }
    return i;
}
const int N=2005;
const int mod=1e9+9;
int n,k;
ll a[N],b[N],C[N][N],f[N][N],nxt[N],mul[N];
inline void pre()
{
    for(int i=0;i<=n;++i) C[i][i]=C[i][0]=1;
    for(int i=1;i<=n;++i)
        for(int j=1;j<i;++j) C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
    mul[1]=1;
    for(int i=2;i<=n;++i) mul[i]=mul[i-1]*i%mod;
}
int main()
{
    freopen("lx.in","r",stdin);
    n=read(),k=read();
    if((n+k)&1)
    {
        cout<<"0"<<endl;
        return 0;
    }
    k=(n+k)>>1;pre();
    for(int i=1;i<=n;++i) a[i]=read();
    for(int i=1;i<=n;++i) b[i]=read();
    sort(a+1,a+n+1);sort(b+1,b+n+1);
    for(int i=1;i<=n;++i)
    {
        int x=a[i];
        x=lower_bound(b+1,b+n+1,x)-b-1;
        nxt[i]=x;
    }

    for(int i=0;i<=n;++i) f[i][0]=1;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
            f[i][j]=(f[i-1][j]+f[i-1][j-1]*(nxt[i]-(j-1))%mod)%mod;

    ll ans=0,delt=1;
    for(int i=k;i<=n;++i,delt=-delt)
        ans=(ans+delt*(f[n][i]*C[i][k]%mod*mul[n-i])%mod)%mod;
    cout<<(ans%mod+mod)%mod<<endl;
}