bzoj千题计划270：bzoj4559: [JLoi2016]成绩比较

Posted 2020-10-27 日拱一卒功不唐捐

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了bzoj千题计划270：bzoj4559: [JLoi2016]成绩比较相关的知识，希望对你有一定的参考价值。

http://www.lydsy.com/JudgeOnline/problem.php?id=4559

f[i][j] 表示前i门课，有j个人没有被碾压的方案数

g[i] 表示第i门课，满足B神排名的分数安排方案数

g[i]的求法：

枚举B神这门课x分，则有n-Ri个人的分数<=x ，Ri-1个人的分数>x

Ui 上限是1e9，但是g[i] 是一个关于Ui 的n次多项式，所以可以用拉格朗日插值法来求

递推 f[i][j]：

假设f[i-1][w] 转移到了f[i][j]，j>=w

前i-1门课没有被碾压，前i门课也一定没有被碾压

前i-1门课被碾压，前i门课可能继续被碾压，也可能不再被碾压

单看这一门课有Ri-1个人的成绩比B神高

但这Ri-1个人之前可能就有科目比B神高，已经不被碾压，这次成绩比B神高还是低都行

所以实际新增加了j-w个没有被碾压的，即这j-w个人这一门的成绩比B神高，之前的科目都比B神低

在已经没有被碾压的w个人中，还存在 Ri-1-（j-w）个人的成绩比B神高

之前有n-w-1个人被碾压，所以新增情况的方案数为C（n-w-1，j-w）

后一种情况的方案数为C（w，Ri-1-j+w）

#include<cstdio>
#include <iostream>

using namespace std;

const int mod=1e9+7;

#define N 101

int C[N][N];
int U[N],rk[N];
int f[N][N];
int g[N];

void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*10+c-\'0\'; c=getchar(); }
}

int get_C(int n,int m)
{
    if(n<0 || m<0 || n<m) return 0;
    return C[n][m];
}

void pre_C()
{
    C[0][0]=1;
    for(int i=1;i<=100;++i)
    {
        C[i][0]=1;
        for(int j=1;j<=i;++j)
            C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
    }
}

int Pow(int a,int b)
{
    int res=1;
    for(;b;a=1LL*a*a%mod,b>>=1)
        if(b&1) res=1LL*res*a%mod;
    return res;
}

int Langrange(int n,int r,int k)
{
    for(int u=1;u<=k;++u)
    {
        g[u]=0;
        for(int x=1;x<=u;++x)
            g[u]=(g[u]+1LL*Pow(u-x,r-1)*Pow(x,k-1-r)%mod)%mod;
        if(n==u) return g[u];
    }
    int fz=1;
    for(int i=1;i<=k;++i) fz=1LL*fz*(n-i)%mod;
    int fm,ans=0;
    for(int i=1;i<=k;++i)
    {
        fm=n-i;
        for(int j=1;j<=k;++j)
            if(i!=j) fm=1LL*fm*(i-j)%mod;
        ans=(ans+1LL*fz*g[i]%mod*Pow(fm,mod-2)%mod)%mod;
    }
    if(ans<0) ans+=mod;
    return ans;
}

int main()
{
    int n,m,k;
    read(n); read(m); read(k);
    for(int i=1;i<=m;++i) read(U[i]);
    for(int i=1;i<=m;++i) read(rk[i]);
    pre_C();
    int G;
    f[0][0]=1;
    for(int i=1;i<=m;++i)
    {
        G=Langrange(U[i],rk[i],n+1);
        for(int j=0;j<=n;++j)
        {
            for(int w=0;w<=j;++w)
                f[i][j]=(f[i][j]+1LL*f[i-1][w]*get_C(w,rk[i]-1-j+w)%mod*get_C(n-w-1,j-w)%mod)%mod;        
            f[i][j]=1LL*f[i][j]*G%mod;
        }
    }
    printf("%d",f[m][n-k-1]);
    return 0;
}