2021牛客暑期多校训练营10 Diameter Counting

题目

分析

cjb的题解已经很详细了。下面是本人理解。
算法的思想是从一个核心开始往外扩展，用$f_{i,j}$表示有$j$个叶子的$i$标号树的个数。然后加叶子时枚举增加的叶子$k$即可转移到$f_{i+k,k}$上了。接着再引入$g_{i,j,k}$表示新加入的叶子的序列方案。于是可以知道如何转移了
$$f_{i+k,k}=f_{i,j}\cdot g_{i,j,k}\cdot C_{i+k}^i$$
然后就是求$g$数组了。
对于数组中的第$k$个节点，如果他是目前没有子节点的旧叶节点，那么就有$i$种方案，即$i\cdot g_{i,j,k-1}$。如果不是，那么方案数即为$j\cdot g_{i-1,j-1,k-1}$。
有了$f,g$数组，最后就是求答案了。这里引入$h_{i,j}$表示有$j$个叶子的$i$标号树的直径和。
将$h_{i,j}$转移到$h_{i+k,i}$要注意新加的节点自身的贡献以及对老树的贡献。于是可以得到
$$h[i+k][k]=h[i+k][k]+h[i][j]\ast g[i][j][k]\ast C[i+k][i]+2\ast f[i][j]\ast C[i+k][k]\ast g[i][j][k]$$
最后
$$ans=\sum _{i=1}^n{h}_{n,i}$$
时间复杂度是$O(n^3)$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,p,ans;
ll h[505][505],f[505][505],g[505][505][505];
ll C[505][505];
int main()
{
	scanf("%lld%lld",&n,&p);A
	for (ll i=0;i<=500;i++) C[i][0]=C[i][i]=1;
	for (ll i=2;i<=500;i++)
	 for (ll j=1;j<i;j++)
	  C[i][j]=(C[i-1][j]+C[i-1][j-1])%p;
	g[0][0][0]=1;
	for (ll i=1;i<=n;i++)
	{
		ll sum=1;
		for (ll k=0;k<=n;k++)
		 {
		 	g[i][0][k]=sum;
		 	sum=sum*i%p;
		 }
	}
	for (ll i=1;i<=n;i++)
	 for (ll j=1;j<=i;j++)
	  for (ll k=j;k<=n-i;k++)
	   g[i][j][k]=(i*g[i][j][k-1]%p+j*g[i-1][j-1][k-1]%p)%p;
	f[1][1]=f[2][2]=1;
	for (ll i=1;i<=n;i++)
	 for (ll j=1;j<=i;j++)
	  for (ll k=j+(i==1?1:0);k<=n-i;k++)
	   f[i+k][k]=(f[i+k][k]+f[i][j]*g[i][j][k]%p*C[i+k][i]%p)%p;
	h[2][2]=1;
	for (ll i=1;i<=n;i++)
	 for (ll j=1;j<=i;j++)
	  for (ll k=j+(i==1?1:0);k<=n-i;k++)
	   h[i+k][k]=((h[i+k][k]+h[i][j]*g[i][j][k]%p*C[i+k][i]%p)%p+2*f[i][j]%p*C[i+k][k]%p*g[i][j][k]%p)%p;
	for (ll i=1;i<=n;i++)
	 ans=(ans+h[n][i])%p;
	printf("%lld\n",ans);
}

ps:记得取模。