acm-(组合计数、dp计数)2020 China Collegiate Programming Contest Qinhuangdao Site H. Holy Sequence

传送门
首先考虑对于数字$j$而言，如何计算贡献，假设它有$x$个，那么$x^2$等价于从整个序列中可以重复地选择两个数字$j$的选法。而所有$x^2$之和其实就是从所有可能的序列中重复地选择两个数字$j$的选法之和。

总的来说，两个选法不同，要么两个选法中选择的两个位置存在差异，要么两个选法对应的序列存在差异，这样的话两个选法都会产生一次贡献。

于是对于数字$j$而言，我们枚举它第一次出现在序列中的位置$i$，设$pre[i][j]$表示长度为$i$的序列，并且$j$仅仅存在于序列元素最后一个的合法方案数，$suf[i][j]$表示长度为$i$的序列，并且$j$作为第一个元素的合法方案数(不考虑$a_0$的限制)。那么对于一个固定的$j,i$而言，我们考虑将所有的选法划分为四类：

1. 两次都选择位置$i$，方案数为$pre[i][j]\cdot suf[n-i+1][j]$。
2. 一次选择位置$i$，另一次选择位置$k,(k>i)$，方案数为$2(n-i)pre[i][j]\cdot suf[n-i][j]$(乘二是因为可以第一次选$i$或第二次选$i$)
3. 两次选择位置$k,(k>i)$，方案数为$pre[i][j]\cdot suf[n-i][j]$
4. 一次选择位置$k,(k>i)$，另一次选择位置$h,(h>i,h\ne k)$，方案数为$2C_{n-i}^{2}pre[i][j]\cdot suf[n-i-1][j]$。

将上述四种情况加起来就是对于一个固定$j$和$i$而言产生的总贡献，枚举$i$从$1\sim n$就能求出对于数字$j$而言的总贡献。

下面考虑$pre[i][j]$和$suf[i][j]$如何求解，设$f[i][j]$表示长度为$i$的序列最大值为$j$的合法序列数，那么不难得出$f[i][j]=f[i-1][j-1]+f[i-1][j]\cdot j$，然后可以发现的是$pre[i][j]=f[i-1][j-1]$，于是有$pre[i][j]=pre[i-1][j-1]+pre[i-1][j]\cdot (j-1)$。

再看$suf[i][j]$，考虑第二个位置上的数与$j$的大小关系：

1. 不大于$j$，此时我们发现第二个位置对后面的位置填什么数字其实没有影响，也就是说我们去掉位置二后，一定存在$suf[i-1][j]$种方案数，加上位置二，并固定位置二上的数字，仍然存在$suf[i-1][j]$种方案，不过由于位置二上可以选择$1\sim j$，故总方案为$suf[i-1][j]\cdot j$。
2. 大于$j$，只可能是$j+1$，此时第一个位置对后面位置填什么数字没有影响，可以去掉，方案数即为$suf[i-1][j+1]$。

将上面两种情况加起来就能得到：$suf[i][j]=suf[i-1][j+1]+suf[i-1][j]\cdot j$。

int n,pre[maxn][maxn],suf[maxn][maxn];

int main(){
	int t=rd(),kase=0;
	while(t--){
		n=rd(),mod=rd();
		pre[1][1]=1;
		FOR(i,2,n+1)
		FOR(j,2,i+1)pre[i][j]=sum(pre[i-1][j-1],1ll*pre[i-1][j]*(j-1)%mod);
		FOR(i,1,n+1)suf[1][i]=1;
		FOR(i,2,n+1)
		ROF(j,n-1,1)suf[i][j]=sum(suf[i-1][j+1],1ll*suf[i-1][j]*j%mod);
		printf("Case #%d:\n",++kase);
		FOR(j,1,n+1){
			int ans=0;
			FOR(i,j,n+1){
				add(ans,1ll*pre[i][j]*suf[n-i+1][j]%mod);
				add(ans,3ll*pre[i][j]*(n-i)%mod*suf[n-i][j]%mod);
				if(i<n)add(ans,1ll*(n-i)*(n-i-1)%mod*pre[i][j]%mod*suf[n-i-1][j]%mod);
			}
			wr(ans),putchar(' ');
		}
		puts("");
	}
}