牛客多校10-D Diameter Counting (图论-求所有包含 n 个（有标记）点的树的直径之和)

题目

题意：给一个数 n ，求所有包含 n 个（有标记）点的树的直径之和。

分析

• 关于直径，有这样一个求法：要求一个树的直径，可以这样一轮轮删点直到剩下一个点或没有点：每一轮把它所有的叶子节点删掉。这样，删了 x 轮，直径就是 2x 或 2x+1。

• 那么我们反过来，要构造一个直径为 d 的树，也可以一轮轮加：只要每一轮加的点都满足上每个叶子节点上都至少加了一个点，这样每一轮过后直径都加了2（除了往单独一个点上加一个点这种极端情况）。于是就有下面的dp思路。

• f[i][j] 代表 i 个（有标记）点，有 j 个叶子节点的树的个数。

• g[i][j][k] 代表从 i 个（有标记）点中可重复地选取 k 个，其中已知的 j 个必须要选取，取完 k 个之后进行排列所得的排列个数。用于辅助dp。

• h[i][j] 代表 i 个（有标记）点，有 j 个叶子节点的树的直径之和。

• 在已知 f[i][j] 的情况下，加一轮， k 个点(k>=j)，则有递推式子 f[i+k][k] = f[i][j] * g[i][j][k] * C[i+k][k]。它的意义是：对于每个 (i,j) 的树，和新的 k 个点，让 g 对应的那个序列依次对应到 k 个点上，然后连到树上。那个组合数是选 i+k 个点中哪些点作为原来的树，哪些点作为新加的叶子节点。

• g的递推则是考虑序列最后一个数选什么。可以看看代码

• h的递推也类似 f。

• 时间复杂度是 n 的三次方

注意

我写的时候好多边界的问题调了好久……

代码

#include <bits/stdc++.h>
using namespace std;

#define MAXN (505)

int n;
long long Ha;
long long ans;
long long C[MAXN][MAXN];
long long f[MAXN][MAXN];
long long g[MAXN][MAXN][MAXN];
long long h[MAXN][MAXN];


void solve()
{
	int i,j,k;
	scanf("%d%lld",&n,&Ha);
	
	for (int i=0;i <=n; i++) C[i][0]=1;
	for (i=1; i<=n; i++)
		for (j=1; j<=i; j++)
			C[i][j]=(C[i-1][j-1]+C[i-1][j])%Ha;
	

	g[0][0][0]=1;
	for (i=1; i<=n; i++) {
		for (int k=g[i][0][0]=1; k<=n; k++)
			g[i][0][k]=g[i][0][k-1]*i%Ha;
		for (j=1; j<=i; j++)
			for (k=1; k<=n; k++)
				g[i][j][k]= ( i*g[i][j][k-1] %Ha +j*g[i-1][j-1][k-1] %Ha )%Ha;
	}
	
	
	f[0][0]=f[1][1]=f[2][2]=1;
	h[2][2]=1;
	
	for (i=2; i<=n; i++) {
		if (i>=3) {
			f[i][i-1]=i;
			h[i][i-1]=2*i;
		}
		for (j=2; j<=i; j++)
			for (k=j; i+k<=n; k++) {
				f[i+k][k]+= f[i][j]*g[i][j][k] %Ha *C[i+k][k] %Ha;
				f[i+k][k]%=Ha;
				
				
				h[i+k][k]+= ( h[i][j]*g[i][j][k] %Ha + 2*f[i][j]*g[i][j][k] %Ha) *C[i+k][k] %Ha;
				h[i+k][k]%=Ha;
			}
	}

	ans=0;
	for (k=1; k<=n; k++) ans=(ans+h[n][k])%Ha;
	printf("%lld\n",ans);
}


int main()
{
	solve();
}