JZOJ2904. 【集训队互测 2012】Calc

题意：

一个序列$a_1,...,a_n$ 是合法的，当且仅当：
长度为给定的$n$。
$a_1,...,a_n$ 都是[1,A]中的整数。
$a_1,...,a_n$ 互不相等。
一个序列的值定义为它里面所有数的乘积，即$a_1{a}_2...a_n$。
求所有不同合法序列的值的和。
两个序列不同当且仅当他们任意一位不一样。
输出答案对一个数$mod$取余的结果。

数据范围：

$0:A\le 10,n\le 10.$
$1..3:A\le 1000,n\le 20.$
$4..9:A\le 10^9,n\le 20.$
$10..19:A\le 10^9,n\le 500.$
$mod\le 10^9,mod$ 为素数，$mod>A>n+1$

Analysis：

首先解决互不相同，不妨将最后序列排序，从小到大统计方案再乘上$n!$即可。
于是我们可以设$f_{i,j}$，表示当前考虑到$[1..i]$，选了$j$个数的方案。
可以$O(nA)$转移，每次考虑当前$i$选不选。能通过前$4$档部分分。
考虑优化转移，套路地去想能不能从$f_{A,i}$推到$f_{2\ast A,i}$。
我们考虑如此一个式子：$$\prod (a_i+A)=\sum _{S\subset a_0..a_n}{A}^{n-|S|}\ast \prod _{a_i\in S}{a}_i$$
通过这个式子我们也许就有了从$A$推到$2A$的方法了。
我们考虑设在$[1,A]$里选$i$个数的价值和为$a_i$
在$[A+1..2A]$选$i$个数的价值和为$b_i$
那么有：$f_{2A,i}={\sum }_{j=0}^i{a}_j\ast b_{i-j}$
怎么算$b_i$?用刚刚那个式子
$$b_i=\sum _{j=0}^i{A}^{i-j}\ast a_i\ast {(}_{i-j}^{A-i})$$
表示枚举那个子集，那么后面的乘积部分可以用$a_i$表示出来，前面的系数是$a_i$乘上有多少个这样的子集，也就是那个组合数。
组合数我们一般都可以用下降幂的形式表示出来。
设$m_i={\prod }_{j=0}^{i-1}(A-j),n{m}_i=m_i^{-1}$。那么它就是：$(i-j){!}^{-1}\ast m_i\ast n{m}_j$。
现在$b_i$就可以算出来了。
我们考虑当$A$为奇数时，递归$A-1$，然后$O(n)$递推到$A$。
为偶数直接递归$A/2$，$O(n^2)$变换到$A$。
复杂度$O(n^2\log A)$。

Code：

# include<cstdio>
# include<cstring>
# include<algorithm>
using namespace std;
const int N = 500 + 5;
typedef long long ll;
int f[65][N],inv[N],fac_[N],c[N];
int mi[N],nmi[N],b[N],a[N];
int n,A,mo;
inline int pow(int x,int p)
{
	int ret = 1;
	for (; p ; p >>= 1,x = (ll)x * x % mo)
	if (p & 1) ret = (ll)ret * x % mo;
	return ret;
}
inline void calc(int B,int d)
{
	if (B == 1) { f[d][0] = f[d][1] = 1; return; }
	if (B & 1)
	{
		calc(B - 1,d + 1); int A = B; f[d][0] = 1;
		for (int i = 1 ; i <= n ; ++i) f[d][i] = (f[d + 1][i] + (ll)f[d + 1][i - 1] * A % mo) % mo;
	}else
	{
		calc(B >> 1,d + 1); int A = B >> 1;
		for (int i = 0 ; i <= n ; ++i) a[i] = f[d + 1][i];
		mi[0] = nmi[0] = c[0] = 1;
		for (int i = 0 ; i < n ; ++i) mi[i + 1] = (ll)mi[i] * (A - i) % mo;
		for (int i = 1 ; i <= n ; ++i) c[i] = (ll)c[i - 1] * A % mo,nmi[i] = pow(mi[i],mo - 2);
		for (int i = 0 ; i <= n ; ++i)
		{
			b[i] = 0;
			for (int j = 0 ; j <= i ; ++j) b[i] = (b[i] + (ll)c[i - j] * a[j] % mo * nmi[j] % mo * fac_[i - j] % mo) % mo;
			b[i] = (ll)b[i] * mi[i] % mo;
		}
		for (int i = 0 ; i <= n ; ++i)
			for (int j = 0 ; j <= i ; ++j) f[d][i] = (f[d][i] + (ll)a[j] * b[i - j] % mo) % mo;
	}
}
int main()
{
	scanf("%d%d%d",&A,&n,&mo),inv[1] = fac_[0] = fac_[1] = 1;
	for (int i = 2 ; i <= n ; ++i)
		inv[i] = (ll)(mo - mo / i) * inv[mo % i] % mo,fac_[i] = (ll)fac_[i - 1] * inv[i] % mo;
	calc(A,0); printf("%d\n",(ll)f[0][n] * pow(fac_[n],mo - 2) % mo);
	return 0;
}