BZOJ - 3622：已经没有什么好害怕的了 （广义容斥）

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

Description

Input

Output

Sample Input

4 2
5 35 15 45
40 20 10 30

Sample Output

4

HINT

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

题意：给定a数组和b数组，大小都为N，现在让你两两配对，使得a>b个个数=(N+K)/2; a<b的个数=(N-K)/2;

思路：用容斥来求。  我们假设a>b为A情况，a<b为B情况。先让ab数组分别排序； f[i][j]表示前i个a数组至少存在j个A情况的方案数，那么可以得到f的递推式。 最后用容斥来累加答案。

推荐CQZhangYU的博客。

#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=;
const int Mod=1e9+;
int C[maxn][maxn],f[maxn][maxn],a[maxn],b[maxn],jc[maxn],ans;
int main()
{
    int N,K; scanf("%d%d",&N,&K);
    jc[]=; rep(i,,N) jc[i]=(ll)jc[i-]*i%Mod;
    rep(i,,N){
        C[i][]=;
        rep(j,,i) C[i][j]=(C[i-][j]+C[i-][j-])%Mod;
    }
    rep(i,,N) scanf("%d",&a[i]);
    rep(i,,N) scanf("%d",&b[i]);
    sort(a+,a+N+) ;sort(b+,b+N+);
    f[][]=;
    rep(i,,N){
        int pos=upper_bound(b+,b+N+,a[i])-b; pos--;
        rep(j,,i){
            f[i][j]=(f[i-][j]+(ll)f[i-][j-]*max(pos-j+,)%Mod)%Mod;
        }
        f[i][]=f[i-][];
    }
    if((N+K)%==) return puts(""),;
    K=(N+K)/;
    rep(i,K,N){
        f[N][i]=(ll)f[N][i]*jc[N-i]%Mod;
        if((i-K)&) ans=((ans-(ll)f[N][i]*C[i][K]%Mod)%Mod+Mod)%Mod;
        else ans=(ans+(ll)f[N][i]*C[i][K]%Mod)%Mod;
    }
    printf("%lld\n",(ans%Mod+Mod)%Mod);
    return ;
}