题目大意:
一个长度为
合法序列
,满足
,且互不相等,价值为序列中全部数的积。有一个位置不同的两个序列不同,求所有满足的序列的价值总和。
且为质数
。
分析:
小的数据显然可以
,设
为选了
个数,且不大于
的总和。显然有
然后发现 很大,然后我就往矩阵乘法的方向想,然后就想不到了。
可以使用某些奇怪的方法知道, 是一个关于 的多项式,他的次数是 。
于是我们预处理出 ,然后拉格朗日插值就可以了。
代码:
// 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));
}