Description
一个序列a1,...,an是合法的,当且仅当:
长度为给定的n。
a1,...,an都是[1,A]中的整数。
a1,...,an互不相等。
一个序列的值定义为它里面所有数的乘积,即a1a2...an。
求所有不同合法序列的值的和。
两个序列不同当且仅当他们任意一位不一样。
输出答案对一个数mod取余的结果。
Solution
由于 (f(A)) 是一个关于 (A) 的 (n) 次多项式 .
知道 (f[1]...f[2*n+1]) 然后插值一下就行了.
(f[1]...f[2*n+1]) 可以 (DP) 求出 , 强制序列递增 , 最后乘以 (n!) 就行了.
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int A,n,m,mod,f[N][N],inv[N],Fac[N];
int main(){
freopen("pp.in","r",stdin);
freopen("pp.out","w",stdout);
cin>>A>>n>>mod;
m=2*n+1;
for(int i=1;i<=m;i++)f[1][i]=i;
for(int i=2;i<=n;i++){
int sum=0;
for(int j=i-1;j<=m;j++){
f[i][j]=1ll*j*sum%mod;
sum=(sum+f[i-1][j])%mod;
}
}
inv[0]=inv[1]=Fac[0]=1;
for(int i=2;i<=m;i++)inv[i]=(mod-1ll*(mod/i)*inv[mod%i]%mod)%mod;
for(int i=1;i<=m;i++)Fac[i]=1ll*Fac[i-1]*i%mod;
for(int i=n;i<=m;i++)f[n][i]=1ll*f[n][i]*Fac[n]%mod;
for(int i=n;i<=m;i++)f[n][i]=(f[n][i]+f[n][i-1])%mod;
if(A<=m)cout<<f[n][A],exit(0);
int ans=0;
for(int i=n;i<=m;i++){
int t=1;
for(int j=1;j<=m;j++)
if(i!=j)t=1ll*t*(A-j)%mod*(i>=j?inv[i-j]:-inv[j-i])%mod;
ans=(ans+1ll*f[n][i]*t)%mod;
}
cout<<(ans+mod)%mod;
return 0;
}