[bzoj4559][DP][拉格朗日插值]成绩比较

Description

G系共有n位同学，M门必修课。这N位同学的编号为0到N-1的整数，其中B神的编号为0号。这M门必修课编号为0到M-
1的整数。一位同学在必修课上可以获得的分数是1到Ui中的一个整数。如果在每门课上A获得的成绩均小于等于B获
得的成绩，则称A被B碾压。在B神的说法中，G系共有K位同学被他碾压（不包括他自己），而其他N-K-1位同学则没
有被他碾压。D神查到了B神每门必修课的排名。这里的排名是指：如果B神某门课的排名为R，则表示有且仅有R-1
位同学这门课的分数大于B神的分数，有且仅有N-R位同学这门课的分数小于等于B神（不包括他自己）。我们需要
求出全系所有同学每门必修课得分的情况数，使其既能满足B神的说法，也能符合D神查到的排名。这里两种情况不
同当且仅当有任意一位同学在任意一门课上获得的分数不同。你不需要像D神那么厉害，你只需要计算出情况数模1 0^9+7的余数就可以了。

Input

第一行包含三个正整数N,M,K，分别表示G系的同学数量（包括B神），必修课的数量和被B神碾压的同学数量。第二
行包含M个正整数，依次表示每门课的最高分Ui。第三行包含M个正整数，依次表示B神在每门课上的排名Ri。保证1
≤Ri≤N。数据保证至少有1种情况使得B神说的话成立。N<=100,M<=100,Ui<=10^9

Output

仅一行一个正整数，表示满足条件的情况数模10^9+7的余数。

Sample Input

3 2 1

2 2

1 2

Sample Output

10

题解

越来越菜
考虑一个dp设$f[i][j]$表示前$i$门课碾压了$j$个人的方案数
枚举这门课你还要继续碾压多少人
有转移
$$f[i][j]=\sum f[i-1][k]\ast C_k^{k-j}\ast C_{n-1-k}^{R[i]-1-(k-j)}\ast P[i]$$
组合数是让他凑够$R[i]-1$个比他高分的人的
$P[i]$表示在第$i$门课中所有合法的成绩
$$P[i]=\sum j^{n-R[i]}(u_i-j{)}^{R[i]-1}$$
你可以知道对于一个最高成绩$mask$，它的合法数是
$$cnt=\sum j^{n-R[i]}(mask-j{)}^{R[i]-1}$$
发现这是一个$n-1$次多项式
那就可以直接插值了呀
复杂度$O(n^{3}logn)$

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<ctime>
#include<map>
#include<bitset>
#define LL long long
#define mp(x,y) make_pair(x,y)
#define mod 1000000007
using namespace std;
inline int read()
{
	int f=1,x=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void write(LL x)
{
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
}
inline void pr1(int x){write(x);printf(" ");}
inline void pr2(LL x){write(x);puts("");}
LL pow_mod(LL a,LL b)
{
	LL ret=1;
	while(b)
	{
		if(b&1)ret=ret*a%mod;
		a=a*a%mod;b>>=1;
	}
	return ret;
}
LL prefix[210],inv[210];
LL y[210],inv1[210],inv2[210];
LL pre[210],suf[210];
LL C(int n,int m)
{
	if(n<m)return 0;
	return prefix[n]*inv[m]%mod*inv[n-m]%mod;
}
LL polation(int n,LL x)
{
	pre[0]=suf[n+1]=1;
	for(int i=1;i<=n;i++)pre[i]=pre[i-1]*((x-i+mod)%mod)%mod;
	for(int i=n;i>=1;i--)suf[i]=suf[i+1]*((x-i+mod)%mod)%mod;
	LL s2=1;
	for(int i=2;i<=n;i++)s2=s2*(1-i)%mod;
	LL bak=1-n,fir=0,ret=0;
	for(int i=1;i<=n;i++)
	{
		LL s1=pre[i-1]*suf[i+1]%mod*y[i]%mod;
		LL temp=s2;
		ret=(ret+s1*pow_mod((temp+mod)%mod,mod-2)%mod)%mod;
		s2=s2*(++fir)%mod;
		if(bak<0)s2=s2*inv1[-bak]%mod;
		else s2=s2*inv2[bak]%mod;
		bak++;
	}
	return (ret+mod)%mod;
}
int n,m,K;
LL U[210],R[210],f[210][210];
LL ad(LL &x,LL y){x+=y;if(x>=mod)x-=mod;}
int main()
{
	prefix[0]=1;for(int i=1;i<=205;i++)prefix[i]=prefix[i-1]*i%mod;
	inv[205]=pow_mod(prefix[205],mod-2);
	for(int i=204;i>=0;i--)inv[i]=inv[i+1]*(i+1)%mod;
	for(int i=0;i<=205;i++)inv1[i]=pow_mod(-i,mod-2);
	for(int i=0;i<=205;i++)inv2[i]=pow_mod(i,mod-2);
	n=read();m=read();K=read();
	for(int i=1;i<=m;i++)U[i]=read();
	for(int i=1;i<=m;i++)R[i]=read();
	f[0][n-1]=1;
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n+1;j++)
		{
			y[j]=0;
			for(int k=1;k<=j;k++)
				y[j]=(y[j]+pow_mod(k,n-R[i])*pow_mod(j-k,R[i]-1)%mod)%mod;
		}
		LL sum=polation(n+1,U[i]);
		for(int j=K;j<=n-1;j++)
			for(int k=j;k<=n-1;k++)
				if(k-j<=R[i]-1)
					ad(f[i][j],f[i-1][k]*C(k,k-j)%mod*C(n-1-k,R[i]-1-k+j)%mod*sum%mod);
		int cnt;
		cnt++;
 	}
	pr2(f[m][K]);
	return 0;
}