作問:kuronosu1024

元の問題文

整数 NN が与えられます。
NN 個の頂点からなる互いに異なる(ラベルなし、根なし)木の総数を XX とします。 XX998244353998244353 で割った余りを求めてください。

想定エスパー法

NN に対して解の大きさが指数関数的に増加していることより、組み合わせ論的な数と予想できます。
また、N<4N < 4 での値がかなり小さいことより対称性が強いものに焦点を当てると、後は気合で木の個数であることをエスパーします。

解説

分割数DPの要領で解くことができます。(FPSを使えば NN 倍高速ですが、ここでは解説しません)

1) 根付き木の数を数える

まず、NN 頂点の根付き木の総数について考えます

根に部分木を追加していくことを考えると、dp[i]dp[i]ii 頂点の根付き木の総数と定義して愚直的な遷移を次のように考えられます

APiA \in P_i について、( PiP_i は自然数 ii の分割の全体集合) B[j]B[j]AA に含まれる jj の個数として、

dp[i+1]=APijdp[j]HB[j]=APij(dp[j]+B[j]1B[j])dp[i+1]=\sum\limits_{A\in P_i} \prod\limits_j {}_{dp[j]}\mathrm{H}_{B[j]}\\ =\sum\limits_{A\in P_i} \prod\limits_j \binom{dp[j] + B[j] - 1}{B[j]}

これで遷移はできるにはできますが、とりうる AA の種類数( =Pi=|P_i| )は非常に大きな数になるので工夫する必要があります。

(分割とは 例)

例えば i=5i=5 において、頂点数 55 の根付き木の作り方は次の 55 通りに分けられます

  • 根に、大きさがそれぞれ {1,1,1,1}\{1,1,1,1\}である計4つの木をつなげる
  • 根に、大きさがそれぞれ {2,1,1}\{2,1,1\}である計3つの木をつなげる
  • 以下同様に、 {2,2},{3,1},{4}\{2,2\}, \{3,1\}, \{4\}

この {1,1,1,1}\{1,1,1,1\}{3,1}\{3,1\} といった集合を 44 の分割といいます

(例終わり)

dp1[i][j]dp_1[i][j] を、ii 頂点の根付き木の総数であって、根につながる部分木の大きさが全て jj 頂点以下であるものの総数と定義します。(dp[i]=dp1[i][i1]dp[i] = dp_1[i][i-1])
AA のうち最も大きい数 kk を含まれる B[k]B[k] 個全て削除したものを AA' とし、それに対応する BBBB' として分けて考えると

j(dp[j]+B[j]1B[j])=j(dp[j]+B[j]1B[j])(dp[k]+B[k]1B[k])\prod\limits_j \binom{dp[j]+B[j]-1}{B[j]}\\ = \prod\limits_j \binom{dp[j]+B'[j]-1}{B'[j]}\cdot \binom{dp[k] + B[k] - 1}{B[k]}

kk , B[k]B[k] が等しいもの同士をまとめると、(この時、Pi=(Piのうちすべての要素がk1以下であるもの)P'_i = (P_i のうちすべての要素が k-1 以下であるもの) とすると APikB[k]A' \in P'_{i-kB[k]} が成り立つので)まとめた総和は

APikB[k]j(dp[j]+B[j]1B[j])(dp[k]+B[k]1B[k])=dp1[ikB[k]+1][k1](dp[k]+B[k]1B[k]) \sum\limits_{A'\in P'_{i-kB[k]}} \prod\limits_j \binom{dp[j]+B'[j]-1}{B'[j]}\cdot\binom{dp[k] +B[k] - 1}{B[k]}\\ =dp_1[i-kB[k]+1][k-1] \cdot \binom{dp[k] + B[k] - 1}{B[k]}

さらに、同じ kk についてもまとめて

x=1i/kdp1[ikx+1][k1](dp[k]+x1x) \sum\limits_{x=1}^{\lfloor i/k \rfloor} dp_1[i-kx+1][k-1] \cdot\binom{dp[k] +x - 1}{x}

この総和は dp1[i+1][k]dp1[i+1][k1]dp_1[i+1][k]-dp_1[i+1][k-1] に等しく、dp[i+1]dp[i+1] を求めるには、取りうる k,B[k]k, B[k] について上式を求めればよく、その取りうる k,B[k]k, B[k] というのは O(NlogN)O(N \log N) 種類で済みます。
また、2項係数も隣同士であるため dp1[ikx+1][k1](dp[k]+x1x)dp_1[i-kx+1][k-1] \cdot\binom{dp[k] +x - 1}{x} も適切に実装すればそれぞれ O(1)O(1) で計算できます。

よって、 O(N2logN)O(N^2 \log N)dp[N]dp[N] を求めることができ、今回の制約でこれは十分高速です。

2) 根なし木の数を数える

根なし木は根となる頂点を区別しないため、求めた値から重複分を減らす必要があります。重複分を数えるために木の重心を考えます。(根と重心が一致するもののみ数えることで重複なく根なし木の数を数えることができます。)

NN が奇数の時

このとき木の重心は常に 11 個 です。木の重心につながる部分木の大きさは常に N/2N/2 以下であるので、dp1[N][(N1)/2]dp_1[N][(N-1)/2] がそのまま答えになります。

NN が偶数の時

このとき木の重心は常に 11 個か 22 個で、11 個の時は NN が奇数の時と同様に求められます (ただし、 N/2N/2 の部分木があると重心が2個になってしまうので、代わりに N/21N/2 -1 とする必要があります)。

22 個の時は、その間の辺で分割したときちょうど N/2N/2 頂点の二つの木に分けられるので、そのような木は (dp[N/2]+12)\binom{dp[N/2]+1}{2} だけ存在します。

以上より、適切な前計算で必要なDPテーブルを O(N2logN)O(N^2 \log N) で求めることができるため、この問題にACすることができます。