鸽子的糖

题目

题目描述
鸽子 $\alpha$ 最近把 $n$ 根棒棒糖拆了开来，正准备拼到一起…

$\alpha$ 现在十分苦恼，它现在需要把这 $n$ 根糖拆出来的 $n$ 个糖果和 $n$ 根棒给重新组合一下，不过 $\alpha$ 并不知道该怎么组装好看，只能先列出几个条件：

1. 所有糖果和棒都要用上。
2. 每对糖果之间都必须直接或间接连接。
3. 显然对于一个棒全部用上的情况，会形成一个环，$\alpha$ 觉得环上点的数量为 $m$ 比较好。

现在 $\alpha$ 想问的是，有多少种满足上述几个条件的无向图呢？

注意，包在糖果上的包装纸上才有标号，而棒上是没有的。两个方案不同当且仅当对于某一个标号的糖果，在两个方案中与它相连的糖果的标号组成的集合不同，并且我们定义每个点的集合在一开始就有它本身（即形成自环的点不会影响方案数）。

答案对 $998244353$ 取模。

输入格式
输入包括一行两个正整数 $n$ 和 $m$ 。

输出格式
输出一行一个整数，表示方案数。

数据范围与约定
对于所有数据，$1\le m\le n\le 3\times 10^3$ 。

思路

$\mathtt{s}\mathtt{u}\mathtt{b}\mathtt{t}\mathtt{a}\mathtt{s}\mathtt{k}1(m=1\vee m=2)$

发现就是求本质不同的树的数量。设 $f(n)$ 为 $n$ 个点时本质不同的树的数量。显然 $f(1)=1$ 。有递推方程式：

$$f(n)=\sum _{i=1}^{n-1}\left(\genfrac{}{}{0ex}{}{n-2}{i-1}\right)i\cdot f(i)f(n-i)$$

为什么是这样呢？发现我们的做法是 拆掉一条边再重新接上，为了不算重，我们规定把 $1$ 作为树根，断开 $n$ 与其父节点之间的连边。步骤如下：

1. 在除了 $1$ 和 $n$ 以外的 $n-2$ 个点中选 $i-1$ 个点，与 $1$ 一起组成 $i$ 个点的树。方案 $\left(\genfrac{}{}{0ex}{}{n-2}{i-1}\right)f(i)$ 。
2. 剩下的 $n-i$ 个点组成一棵树，以 $n$ 作为根节点。方案数为 $f(n-i)$ 。
3. 选取 $i$ 个点中的任意一个作为 $n$ 的父节点。方案数为 $i$ 。

乘法原理可得该递推式。

---

$2019/11/28\mathtt{u}\mathtt{p}\mathtt{d}\mathtt{a}\mathtt{t}\mathtt{e}$

只需要使用 $\mathtt{p}\mathtt{r}\mathtt{u}\mathtt{f}\mathtt{e}\mathtt{r}$ 序列就可以直接得到 $f_n=n^{n-2}$ 的方便打法——当然也可以用递推的方式验证。

果然我这种dp又差、学的又少的人就会被淘汰。

---

$\mathtt{s}\mathtt{u}\mathtt{b}\mathtt{t}\mathtt{a}\mathtt{s}\mathtt{k}\infty$

模仿上面的想法，发现只改变点的数量，所以考虑用 $g(n)$ 表示 $n$ 个点组成环上有 $m$ 个点的本质不同的基环树的数量（$m$ 即输入中的 $m$ ）。边界条件 $g(m)=\frac{(m-1)!}{2}$ ，递推式

$$g(n)=\frac{n}{n-m}\sum _{i=m}^{n-1}\left(\genfrac{}{}{0ex}{}{n-1}{i}\right)ig(i)f(n-i)$$

初状态 $g(m)$ 就是每个点都在环上。考虑从 $1$ 开始按顺序生成一个序列，有 $(m-1)!$ 种方案。但是顺时针和逆时针实际上是一样的，所以再除以 $2$ 。即：$1-2-3-4$ 和 $1-4-3-2$ 本质是一个环。画一画就知道了！

题外话：我最近在一本数学奥赛、讲组合数学专题的书上看到了这个玩意儿（下图左一）

其他的情况，还是拆掉一条边吧：

• 若 $1$ 不在环上，把环上的点都视做树根，拆掉 $1$ 与其父节点的连边。那么就得从除去 $1$ 的 $(n-1)$ 个点中选 $i$ 个点组成起始基环树，然后把 $1$ 接上去（并不是一定要接到环上去，不然会算掉 $1$ 不与环直接相连的情况）。答案是 $X={\sum }_{i=m}^{n-1}\left(\genfrac{}{}{0ex}{}{n-1}{i}\right)ig(i)f(n-i)$ 。
• $1$ 在环上的情况，将环上的点的点集记作 $S$ ，就 $\mathtt{s}\mathtt{w}\mathtt{a}\mathtt{p}(1,v)(v\in S)$ 一下就行了！所以是 $mX$ 。听上去很合理。

可是 $mX$ 有重复的情况啊！

考虑一个 $1$ 在环上的情况，发现它的“前身”可以是 $1$ 与非环上的 $n-m$ 个点中任意一个点 $\mathtt{s}\mathtt{w}\mathtt{a}\mathtt{p}$ 得到的结果。

比如楼上的方案。它可以是这两种情况：

也就是说，每 $n-m$ 个方案都是一样的！所以只有 $mX÷(n-m)=\frac{m}{n-m}X$ 种情况。

由加法原理，$\mathtt{a}\mathtt{n}\mathtt{s}=\frac{m}{n-m}X+X=\frac{m+(n-m)}{n-m}X=\frac{n}{n-m}X$ 。

代码

#include <cstdio>
#include <iostream>
#include <vector>
using namespace std;

const int MaxN = 3000, Mod = 998244353;
int f[MaxN+5], g[MaxN+5], c[MaxN][MaxN], n, m, inv[MaxN+5];

int main(){
    scanf("%d %d",&n,&m);
    for(int i=(inv[1]=1)+1; i<=n; ++i)
    	inv[i] = (0ll+Mod-Mod/i)*inv[Mod%i]%Mod;
    for(int i=0; i<n; ++i) // 杨辉三角 
        for(int j=c[i][0]=1; j<=i; ++j)
            c[i][j] = (c[i-1][j]+c[i-1][j-1])%Mod;
    f[1] = 1;
    for(int i=2; i<=n; ++i) for(int j=1; j<i; ++j)
        f[i] = (1ll*j*f[j]%Mod*f[i-j]%Mod*c[i-2][j-1]+f[i])%Mod;
    // 或者用f[i] = qkpow(i,i-2,Mod)也可以 
    if(m == 1 or m == 2){
        printf("%d",f[n]);
        return 0;
    }
    g[m] = 1;
    for(int i=3; i<m; ++i) // (m-1)!/2
        g[m] = (1ll*g[m]*i)%Mod;
    for(int i=m+1; i<=n; ++i){
        for(int j=m; j<i; ++j)
            g[i] = (1ll*j*g[j]%Mod*f[i-j]%Mod*c[i-1][j]+g[i])%Mod;
        g[i] = 1ll*g[i]*i%Mod*inv[i-m]%Mod;
    }
    printf("%d",g[n]);
    return 0;
}