CF1439D INOI Final Contests

Description

• 传送门

Solution

• 首先两个段在不相交的时候是互相独立的，可以直接用$f_n,g_n$表示长度$n$的方案数以及答案的和，转移要么是新加点在左边或右边，要么合并两个段，可以$O(n^2)$简单计算。
• 然后直接$O(n^3)$可以把段合并在一起，这样总的时间复杂度就是$O(n^3)$的了。
• 实际上可以做到$O(n^2)$：考虑后面的$n^3$的卷积，我们用生成函数表示出来：
• 设$f(x),g(x)$为$f_n,g_n$的EGF，$F(x)=f(x)+g(x)y$

$$\begin{gathered} \sum _{k\ge 0}{F}^k(x)\frac{u^k}{(1-u{)}^{k+1}}[x^m][u^{n+1-m}] \\ =\frac{1}{1-u}\cdot \frac{1}{1-\frac{F(x)u}{1-u}}=\frac{1}{1-u(F(x)+1)}=\sum _{k\ge 0}{u}^i(F(x)+1{)}^i[x^m][u^{n+1-m}] \\ =(F(x)+1{)}^{n+1-m}[x^m] \\ =(f(x)+g(x)y+1{)}^{n+1-m}[x^m][y] \\ =(n+1-m)g(x)(f(x)+1{)}^{n-m}[x^m] \end{gathered}$$

• 将$(f(x)+1{)}^{n-m}$用一个$n^2$的多项式快速幂（求导，递推）即可做出。

• 还可以继续优化，上面的求快速幂可以多项式exp，$f(x)$通过推式子（OEIS?）可以得到$f_n=2^n(n+1{)}^{n-1}$，$g(x)$可以用分治NTT做到$nlo{g}^{2}n$的复杂度，或者$g(x)$也可以推出一个一阶微分方程套用公式用多项式exp求解？可以做到$nlogn$

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define maxn 505
#define ll long long 
using namespace std;

int n,m,mo,i,j,k;
ll f[maxn],g[maxn],C[maxn][maxn],ans;
ll f0[maxn][maxn],g0[maxn][maxn];
ll P(int n,int m){return C[n+m-1][m-1];}

int main(){
	freopen("ceshi.in","r",stdin);
	scanf("%d%d%d",&n,&m,&mo);
	for(C[0][0]=1,i=1;i<=n+1;i++) for(C[i][0]=1,j=1;j<=i;j++)
		C[i][j]=(C[i-1][j]+C[i-1][j-1])%mo;
	f[0]=1,g[0]=0,f[1]=2,g[1]=0;
	for(i=2;i<=n;i++) {
		(f[i]+=f[i-1]*(i+1)*2)%=mo;
		(g[i]+=f[i-1]*i%mo*(i-1))%=mo;
		(g[i]+=g[i-1]*(i+1)*2)%=mo;
		for(j=1;j+1<i;j++){ 
			k=i-j-1;
			(f[i]+=f[j]*f[k]%mo*(j+k+2)%mo*C[j+k][k])%=mo;
			(g[i]+=(g[j]*f[k]+g[k]*f[j])%mo*(j+k+2)%mo*C[j+k][k])%=mo;
			(g[i]+=f[j]*f[k]%mo*C[j+k][k]%mo*((1ll*(j+1)*j/2+1ll*(k+1)*k/2)%mo))%=mo;
		}
	}
	f0[0][0]=1,g0[0][0]=0;
	for(i=0;i<=m;i++) for(j=0;j<=m;j++) if (f0[i][j]){
		for(k=1;j+k<=m;k++){
			(f0[i+1][j+k]+=f0[i][j]*f[k]%mo*C[j+k][k])%=mo;
			(g0[i+1][j+k]+=(g0[i][j]*f[k]+f0[i][j]*g[k])%mo*C[j+k][k])%=mo;
		}
	}
	for(i=1;i<=m;i++) if (n+1-m-i>=0)
		(ans+=g0[i][m]*P(n+1-m-i,i+1))%=mo;
	printf("%lld\n",ans);
}