[集训队互测]calc

题目

点这里看题目。

分析

首先不难想到可以枚举递增的序列，最后在答案里面乘上\(n!\)，于是有\(O(nk)\)的暴力 DP 一枚：

\(f(i,j)\)表示长度为\(i\)、最大值\(\le j\)的序列的贡献和。

转移显然：

\[f(i,j)=j\times f(i-1,j-1)+f(i,j-1) \]

那么可以发现，当序列长度固定的时候，\(f(n,x)\)肯定是关于\(x\)的函数。环顾四周，DP 转移方程中并不存在除法、开方、作为指数乘方等运算，所以可以推测\(f(n,x)\)就是\(x\)的多项式函数。

那么，它的次数是多少呢？这直接决定了我们如何进行插值。设\(f(n,x)\)的次数为\(g(n)\)，考虑到转移左右两边的次数应该是相等的，就有：

\[f(i,j)-f(i,j-1)=j\times f(i-1,j-1)\Rightarrow g(n)-1=g(n-1)+1 \]

补充一下，多项式函数做差分，即\(f(x)-f(x-1)\)，得到的结果的次数会比原多项式的小一，可以直接用二项式定理展开证明。

然后发现，\(g(n)=g(n-1)+2\)，由于\(f(0,x)=1\)，所以\(g(0)=0\)，得到通项公式\(g(n)=2n\)。

然后我们就知道了\(f(n,x)\)是关于\(x\)的\(2n\)次的多项式函数，因此，我们需要算出\(2n+1\)个点值，用于插值。总时间\(O(n^2)\)。

//f(i,j)=f(i-1,j-1)*j+f(i,j-1)
#include <cstdio>

const int MAXN = 1005;

template<typename _T>
void read( _T &x )
{
	x = 0;char s = getchar();int f = 1;
	while( s > '9' || s < '0' ){if( s == '-' ) f = -1; s = getchar();}
	while( s >= '0' && s <= '9' ){x = ( x << 3 ) + ( x << 1 ) + ( s - '0' ), s = getchar();}
	x *= f;
}

template<typename _T>
void write( _T x )
{
	if( x < 0 ){ putchar( '-' ); x = ( ~ x ) + 1; }
	if( 9 < x ){ write( x / 10 ); }
	putchar( x % 10 + '0' );
}

int f[MAXN][MAXN], y[MAXN];
int N, K, M, mod;

int qkpow( int base, int indx )
{
	int ret = 1;
	while( indx )
	{
		if( indx & 1 ) ret = 1ll * ret * base % mod;
		base = 1ll * base * base % mod, indx >>= 1;
	}
	return ret;
}

int inver( const int a ) { return qkpow( a, mod - 2 ); }
void add( int &x, const int v ) { x = ( x + v >= mod ? x + v - mod : x + v ); }

int Lagrange()
{
	if( K <= M ) return y[K];
	int ans = 0, tmp;
	for( int i = 1 ; i <= M ; i ++ )
	{
		tmp = 1;
		for( int j = 1 ; j <= M ; j ++ )
			if( i != j )
				tmp = 1ll * tmp * ( K - j ) % mod * inver( i - j + mod ) % mod;
		add( ans, 1ll * tmp * y[i] % mod );
	}
	return ans;
}

int main()
{
	read( K ), read( N ), read( mod );
	M = 2 * N + 1;
	for( int j = 0 ; j <= M ; j ++ )
		f[0][j] = 1; 
	for( int i = 1 ; i <= N ; i ++ )
		for( int j = 1 ; j <= M ; j ++ )
			f[i][j] = ( 1ll * f[i - 1][j - 1] * j % mod + f[i][j - 1] ) % mod;
	for( int i = 0 ; i <= M ; i ++ ) y[i] = f[N][i];
	int fac = 1; 
	for( int i = 1 ; i <= N ; i ++ ) fac = 1ll * fac * i % mod;
	write( 1ll * fac * Lagrange() % mod ), putchar( '\n' );
	return 0;
}