1. 程式人生 > >bzoj 4559 [JLoi2016]成績比較——拉格朗日插值

bzoj 4559 [JLoi2016]成績比較——拉格朗日插值

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=4559

關於拉格朗日插值,可以看這些部落格:

https://www.cnblogs.com/ECJTUACM-873284962/p/6833391.html

https://blog.csdn.net/qq_35649707/article/details/78018944

這個題要先想好DP方程。dp[ i ][ j ]表示第 i 門課、目前有 j 個人被“碾壓”。

dp[ i ][ j ]=sigma( dp[ i-1 ][ k ] * C( k , j ) * C( n-k-1 , n-rk[ i ]-j ) * g[ i ] )。其中 g[ i ]=sigma(d=1~u[ i ]) d^(n-rk[ i ]) * ( u[ i ]-d )^(rk[ i ]-1)。

C( k , j )表示從上一次的 k 個人裡選 j 個人作為這次還是分數<=自己的人;第二個C就是在已經不被碾壓的人中選一些滿足自己的排名。

然後每個人的分數是在自己之上還是在自己之下就確定了。列舉自己的分數,在自己之下的人每個有 d 種選擇,在自己之上的人每個有 ( u[ ] - d ) 種選擇。

g用拉格朗日插值算就行。是一個 n-1 次函式。但不知為何需要 n+1 個點才行。

注意 upd( ) 裡寫上 & !!!!!

負數的逆元果然可以是它相反數逆元的相反數。

最後是恰好 K 個人,不是大於等於 K 個人。

注意 g 是sigma的,不是單獨的 d^( ) * ( )^( ) 。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=105,mod=1e9+7;
int n,m,t,u[N],rk[N],dp[N][N],c[N][N],g[N],y[N],inv[N],ans;
void upd(int &x){x>=mod?x-=mod:0;x<0?x+=mod:0;}//////&!!!!!!!
int
pw(int x,int k) { int ret=1;while(k){if(k&1)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=1;}return ret; } int calc(int lm,int n,int m) { int ret=0,d=n+m+2;//+2? if(lm<=d) { for(int i=1;i<=lm;i++) ret=(ret+(ll)pw(i,n)*pw(lm-i,m))%mod; return ret; } for(int i=1;i<=d;i++) { y[i]=(y[i-1]+(ll)pw(i,n)*pw(lm-i,m))%mod;//y[i-1]+* !! int tmp=1; for(int j=1;j<=d;j++) { if(j==i)continue; if(i>j) tmp=(ll)tmp*(lm-j)%mod*inv[i-j]%mod; else tmp=(ll)tmp*(j-lm)%mod*inv[j-i]%mod; } ret=(ret+(ll)tmp*y[i])%mod; } return ret; } void init() { 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],upd(c[i][j]); for(int i=1;i<=n;i++)inv[i]=pw(i,mod-2); for(int i=1;i<=m;i++) g[i]=calc(u[i],n-rk[i],rk[i]-1); } int main() { scanf("%d%d%d",&n,&m,&t); for(int i=1;i<=m;i++)scanf("%d",&u[i]); for(int i=1;i<=m;i++)scanf("%d",&rk[i]); init(); dp[0][n-1]=1; int mn0=n-1,mn1=n-1; for(int i=1;i<=m;i++) { mn1=min(mn0,n-rk[i]); for(int j=t;j<=mn1;j++)//t { for(int k=j;k<=mn0;k++) if(dp[i-1][k]) dp[i][j]=(dp[i][j]+(ll)dp[i-1][k]*c[k][j]%mod*c[n-1-k][n-rk[i]-j]%mod*g[i])%mod; } mn0=mn1; } printf("%d\n",dp[m][t]); return 0; }