【JZOJ4924】向再见说再见

Description

Solution

首先把A、B排序。

设$f_{i,j}$表示做到A队第$i$个人，有$j$个人赢的方案数。

如果第$i$个人赢了，那么$f_{i-1,j-1}\times (can-(j-1))\longrightarrow f_{i,j}$（$can$为第$i$个人能赢的人数）。

如果第$i$个人输了，那么需要往后考虑输给谁，这就很麻烦。

那么我们先不管，也就是$f_{i-1,j}\longrightarrow f_{i,j}$

然后$f_{i,j}$求的就是至少有$j$个人赢的方案数。

设$g_i=f_{n,i}$，$h_i$为恰好赢$i$个人的方案数。

那么有$g_i=\sum_{j\geq i}h_j\cdot C_j^i$
移项：$h_i=g_i-\sum_{j>i}h_j\cdot C_j^i$

然后倒着做去重。

至于这个 C <script type="math/tex" id="MathJax-Element-68">C</script>，可以发现如果用状态表示每个人的匹配，会发现有多种状态能转移到一种状态，于是去掉即可。

注意n与k同奇偶的时候才有解，k=0只要统计一边。

Code

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define N 2010
#define mo 1000000007
#define ll long long
using namespace std;
int a[N],b[N];
ll f[N][N];
ll c[N][N];
ll jc[N];
ll h[N],g[N];
int main()
{
    int n,k;
    scanf("%d %d",&n,&k);
    c[0][0]=1;
    fo(i,1,n)
    {
        c[i][0]=1;
        fo(j,1,i) c[i][j]=(c[i-1][j]+c[i-1][j-1])%mo;
    }
    jc[0]=1;
    fo(i,1,n) jc[i]=jc[i-1]*i%mo;
    fo(i,1,n) scanf("%d",&a[i]);
    fo(i,1,n) scanf("%d",&b[i]);
    sort(a+1,a+n+1);
    sort(b+1,b+n+1);
    int p=0;
    f[0][0]=1;
    fo(i,1,n)
    {
        while(a[i]>b[p+1] && p<n) p++;
        fo(j,0,i)
        {
            f[i][j]=f[i-1][j];
            if(j && p>=j) f[i][j]=(f[i][j]+f[i-1][j-1]*(p-j+1)%mo)%mo;
        }
    }
    fo(i,0,n) g[i]=f[n][i]*jc[n-i]%mo;
    fd(i,n,0)
    {
        h[i]=g[i];
        fo(j,i+1,n) h[i]=(h[i]-h[j]*c[j][i]%mo+mo)%mo;
    }
    if(!k) printf("%lld",h[n/2]);
    else if((n+k)%2==1) printf("0");
    else printf("%lld",(h[(n+k)/2]+h[(n-k)/2])%mo);
}