bzoj3622: 已经没有什么好害怕的了

想一想就是放n^2,先转化一下变成计算至少要(n+k)/2个糖果大于药片

考虑没有重复,先排下序,然后处理出每个a[i]可以和多少b[i]匹配满足条件

本来我的想法就是直接f[i][j]表示枚举到第i位j个满足条件

结果转移不了,改了改变成f[i][j]表示枚举到第i位至少j个满足条件

然后容斥

————————————————————————————————————

upd:之前讲的不知所云。。。

排完序以后处理出d[i]表示第i个糖果可以和1~第d[i]个药片匹配令糖大于药

我们要算的是匹配到第n颗糖,用了n个药片,有k颗糖>它对应的药片,设它为dp[n][n][k]吧

这个东西不好弄,我们容斥曲线救国

设f[i][j]表示匹配到第i颗糖,和j个药片匹配(对应药片可以重复),且糖均大于药的方案数

对于当前f[i][j],假如我用用过的药,方案数就是f[i-1][j],假如用一个没用过的,就有d[i]-(j-1)种选择

f[i][j]=f[i-1][j]+f[i-1][j-1]*(d[i]-(j-1))

f[n][n]相当于搞出了匹配到第n颗糖,用了n个药片,有n颗糖>它对应的药片,也就是dp[n][n][n]

而其实前面两维是没有用的,只用dp[n]表示匹配到第n颗糖,用了n个药片,至少有n颗糖>它对应的药片即可

那么如何通过dp[n]倒着推出dp[n-1]……dp[k]呢?

假设我们现在要求dp[x],大于x的已经搞完了

既然需要正确匹配x个,那么f[n][x]是表示匹配到第n颗糖,用了x药片,全部糖大于药的方案数,假如换种想法,匹配了这x个药片以后,剩下的n-x颗糖随便乱匹配,此时正确匹配数应该是>=x的

所以首先先选n-x颗糖一一放去那些没有用于匹配的药片那,方案数为(n-x)!

所以f[n][x]*(n-x)!就是n颗糖一一对应,且至少有x颗糖比它对应的药片大的方案数了

然后需要减掉包含在里面正确匹配数==x+1的,减去正确==x+2的……,这就是容斥的思想,注意不是加减交互,因为我们已经算出的是刚好有x+1颗糖大于药的方案数了。

所以dp[x]=f[n][x]*(n-x)!-sigema(i>x)dp[i]*C(i,x) 组合数的意思是在i组正确匹配中,选出x个对于当前是组成那x个正确匹配的,剩下i-x是后面乱选不小心搞出来的

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const LL mod=1e9+9;

LL a[2100],b[2100];int d[2100];
LL f[2100][2100],dp[2100];

LL fac[2100],c[2100][2100];
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    int n,k;
    scanf("%d%d",&n,&k);k=(n+k)/2;
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++)scanf("%lld",&b[i]);
    sort(a+1,a+n+1);sort(b+1,b+n+1);
    
    int it=0;
    for(int i=1;i<=n;i++)
    {
        while(a[i]>b[it+1]&&it<n)it++;
        d[i]=it;
    }
    
    memset(f,0,sizeof(f));
    for(int i=0;i<=n;i++)f[i][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=i;j++)
            f[i][j]=(f[i-1][j]+(f[i-1][j-1]*max(d[i]-(j-1),0))%mod)%mod;
            
    fac[0]=1;for(int i=1;i<=n;i++)fac[i]=(fac[i-1]*i)%mod;
    for(int i=0;i<=n;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;
            
    memset(dp,0,sizeof(dp));
    for(int i=n;i>=k;i--)
    {
        dp[i]=(f[n][i]*fac[n-i])%mod;
        for(int j=n;j>i;j--)
            dp[i]=((dp[i]-dp[j]*c[j][i])%mod+mod)%mod;
    }
    printf("%lld\n",dp[k]);
    
    return 0;
}

 

posted @ 2018-09-15 12:53  AKCqhzdy  阅读(199)  评论(0编辑  收藏  举报