题目大意:
一个长度为
n
n
合法序列,满足
∀i,ai∈[1,A]
∀
i
,
a
i
∈
[
1
,
A
]
,且互不相等,价值为序列中全部数的积。有一个位置不同的两个序列不同,求所有满足的序列的价值总和。
(A<=109,n<=500,mod<=109
(
A
<=
10
9
,
n
<=
500
,
m
o
d
<=
10
9
且为质数
)
)
。
分析:
小的数据显然可以,设
f[i][j]
f
[
i
]
[
j
]
为选了
i
i
个数,且不大于的总和。显然有
然后发现 A A 很大,然后我就往矩阵乘法的方向想,然后就想不到了。
可以使用某些奇怪的方法知道,是一个关于 j j 的多项式,他的次数是。
于是我们预处理出 f[n][k],k∈[1,2∗n+1] f [ n ] [ k ] , k ∈ [ 1 , 2 ∗ n + 1 ] ,然后拉格朗日插值就可以了。
代码:
// luogu-judger-enable-o2
#include <iostream>
#include <cstdio>
#include <cmath>
#define LL long long
const int maxn=1007;
using namespace std;
LL n,k;
LL f[maxn][maxn];
LL m,mod;
LL l[maxn],r[maxn],jc[maxn],njc[maxn];
LL power(LL x,LL y)
{
if (y==1) return x;
LL c=power(x,y/2);
c=(c*c)%mod;
if (y%2) c=(c*x)%mod;
return c;
}
LL calc(LL m)
{
l[0]=1;
for (LL i=1;i<=k;i++) l[i]=l[i-1]*(m-i)%mod;
r[k+1]=1;
for (LL i=k;i>0;i--) r[i]=r[i+1]*(m-i)%mod;
jc[0]=1;
for (LL i=1;i<=k;i++) jc[i]=(jc[i-1]*i)%mod;
njc[0]=1;
for (LL i=1;i<=k;i++) njc[i]=(njc[i-1]*(mod-i))%mod;
LL sum=0;
for (int i=1;i<=k;i++)
{
sum=(sum+f[n][i]*l[i-1]%mod*r[i+1]%mod*power(jc[k-i],mod-2)%mod*power(njc[i-1],mod-2)%mod)%mod;
}
return sum;
}
int main()
{
scanf("%lld%lld%lld",&m,&n,&mod);
for (LL j=0;j<=2*n+1;j++) f[0][j]=1;
for (LL i=1;i<=n;i++)
{
for (LL j=1;j<=2*n+1;j++)
{
f[i][j]=(f[i-1][j-1]*i%mod*j%mod+f[i][j-1])%mod;
}
}
k=2*n+1;
if (m<=2*n+1) printf("%lld",f[n][m]);
else printf("%lld",calc(m));
}