【BZOJ】4559: [JLoi2016]成绩比较 计数DP+排列组合+拉格朗日插值

【题意】n位同学（其中一位是B神），m门必修课，每门必修课的分数是[1,Ui]。B神碾压了k位同学（所有课分数<=B神），且第x门课有r_x-1位同学的分数高于B神，求满足条件的分数情况数。当有一位同学的一门必修课分数不同时视为两种情况不同。n,m<=100，Ui<=10^9。

【算法】计数DP+排列组合+拉格朗日插值

【题解】把分数作为状态不现实，只能逐门课考虑。

设$f[i][j]$表示前i门课，有j个同学被碾压的情况数，则有：

$$f[i][j]=g(i)cdotsum_{k=j}^{n}f[i-1][k]cdotinom{k}{k-j}cdotinom{n-k-1}{r_i-1-k+j}$$

解释：首先可以发现当天分数需要高于B神的人数是确定的，和多少人被碾压等信息无关，所以令g(i)表示第i门课的合法分数情况数，独立计算。

枚举前i-1门课被碾压的人数k，那么ri-1由两部分组成，一部分是不再被碾压的k-j人（从k人中选出），剩余的ri-1-k+j人从原本就未被碾压的n-k-1人中选出。

考虑计算g(i)，枚举B神的分数i，则有r-1人的选择范围是[i+1,Ui]，另外n-r人的选择范围是[1,i]，即：

$$g(i)=sum_{i=1}^{U_i}(U_i-i)^{r_i-1}*i^{n-r}$$

Ui太大了，考虑将Ui当成自变量后用拉格朗日插值解决，即：

$$f(x)=sum_{i=1}^{x}(x-i)^{r-1}*i^{n-r}$$

现在我们要求f(Ui)的值，需要确定多项式的次数。网上的解释都看不懂，强行理解：令i=x/2，那么式子右边的i也可以表示为(n-i)，合并后次数为n-1，再加上Σ的上届为x，那么最高次就是n，这是一个n次多项式。

于是我们可以对每个i，O(n^2)枚举前n+1个点的值来插值得到f(Ui)。DP转移的复杂度也是O(n)的。

总复杂度O(n^3)。

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=110,MOD=1e9+7;
int v[maxn],n,m,kind,u[maxn],r[maxn],g[maxn],f[maxn][maxn],c[maxn][maxn];
int power(int x,int k){int ans=1;while(k){if(k&1)ans=1ll*ans*x%MOD;x=1ll*x*x%MOD;k>>=1;}return ans;}
int inv(int x){return power(x,MOD-2);}
int M(int x){return x>=MOD?x-MOD:x;}
int solve(int u,int r){
    for(int x=1;x<=n+1;x++){
        g[x]=0;//!
        for(int i=1;i<=x;i++){
            g[x]=M(g[x]+1ll*power(x-i,r-1)*power(i,n-r)%MOD);
        }
        if(x==u)return g[x];
    }
    for(int i=1;i<=n+1;i++){
        v[i]=1;
        for(int j=1;j<=n+1;j++)if(i!=j)v[i]=1ll*v[i]*(i-j+MOD)%MOD;
        v[i]=inv(v[i]);
    }
    int ans=0;
    for(int i=1;i<=n+1;i++){
        int w=1ll*g[i]*v[i]%MOD;
        for(int j=1;j<=n+1;j++)if(i!=j)w=1ll*w*(u-j+MOD)%MOD;//i!=j
        ans=M(ans+w);
    }
    return ans;
}
int main(){
    scanf("%d%d%d",&n,&m,&kind);
    for(int i=0;i<=n;i++){
        c[i][0]=1;
        for(int j=1;j<=i;j++){
            c[i][j]=M(c[i-1][j-1]+c[i-1][j]);
        }
    }
    for(int i=1;i<=m;i++)scanf("%d",&u[i]);
    for(int i=1;i<=m;i++)scanf("%d",&r[i]);
    f[0][n-1]=1;
    for(int i=1;i<=m;i++){
        int g=solve(u[i],r[i]);
        for(int j=kind;j<=n;j++){
            for(int k=j;k<=n;k++)if(r[i]-1-k+j>=0&&r[i]-1-k+j<=n-k-1){
                f[i][j]=M(f[i][j]+1ll*f[i-1][k]*c[k][k-j]%MOD*c[n-k-1][r[i]-1-k+j]%MOD);
            }
            f[i][j]=1ll*f[i][j]*g%MOD;
        }
    }
    printf("%d",f[m][kind]);
    return 0;
}