bzoj 3622: 已经没有什么好害怕的了 dp+容斥原理

题意

给出两个长度为n的序列，满足任意两个元素互不相同。现在要拿第一个数列的数和第二个数列的数两两匹配，设第一个数列的数大于第二个数列的数的组数为s，问有多少种匹配方案使得s-(n-s)=k
n<=2000

分析

好强啊，根本想不到。。。
题解
先将两个数列排序，设next[i]表示一个最大的j满足a[i]>b[j]，f[i,j]表示匹配了第一个数列的前i个数，有j组是a>b，其余不管的方案数。显然f[i,j]=f[i-1,j]+f[i-1,j-1]*(next[i[]-j+1)
但是这样会算重，我们设g[j]表示全部匹配完后有j组满足a>b，剩下的一定满足b>a的方案数，那么
附一波题解的解释：
(n-i)! 是枚举后面 n-i 可能的方案，g[j]*C(j, i) 表示 f[n,i] 中实际有 j 组药片数 > 糖果数却被计入 f[n,i] 的数量
g[s]就是答案了，总时间复杂度为 O(n^2)

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=2005;
const int MOD=1000000009;

int n,m,a[N],b[N],g[N],f[N][N],next[N],jc[N],ny[N];

int ksm(int x,int y)
{
    int ans=1;
    while (y)
    {
        if (y&1) ans=(LL)ans*x%MOD;
        x=(LL)x*x%MOD;y>>=1;
    }
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<=n;i++) scanf("%d",&b[i]);
    if((n+m)%2==1)
    {
        cout<<0<<endl;
        return 0;
    }
    m=(n+m)/2;
    sort(a+1,a+n+1);sort(b+1,b+n+1);
    for (int i=1;i<=n;i++)
    {
        next[i]=next[i-1];
        while (b[next[i]+1]<a[i]&&next[i]<n) next[i]++;
    }
    jc[0]=ny[0]=1;
    for (int i=1;i<=n;i++) jc[i]=(LL)jc[i-1]*i%MOD,ny[i]=ksm(jc[i],MOD-2);
    f[0][0]=1;
    for (int i=1;i<=n;i++)
        for (int j=0;j<=i;j++)
            f[i][j]=(f[i-1][j]+(LL)f[i-1][j-1]*max(next[i]-j+1,0)%MOD)%MOD;
    for (int i=n;i>=m;i--)
    {
        g[i]=(LL)f[n][i]*jc[n-i]%MOD;
        for (int j=i+1;j<=n;j++) g[i]=(g[i]-(LL)g[j]*jc[j]%MOD*ny[i]%MOD*ny[j-i]%MOD)%MOD;
    }
    printf("%d",(g[m]+MOD)%MOD);
    return 0;
}