atcoder#ABC221F. [ABC221F] Diameter set

[ABC221F] Diameter set

题目描述

N N 頂点からなる木が与えられます。 頂点は 1 1 , 2 2 , \ldots , N N と番号付けられており、 1 i N1 1\leq\ i\leq\ N-1 について、i i 本目の辺は頂点 Ui U_i と頂点 Vi V_i を結んでいます。 木の直径を D D とするとき、木の頂点のうち 2 2 点以上を選んで赤く塗る方法であって、 赤く塗られたどの頂点の間の距離も D D であるようなものの数を 998244353 998244353 で割った余りを求めてください。

ただし、木の 2 2 頂点の間の距離は一方から他方へ移動するときに用いる辺の本数の最小値であり、 木の直径は任意の 2 2 頂点の間の距離の最大値として定められます。

输入格式

入力は以下の形式で標準入力から与えられる。

N N U1 U_1 V1 V_1 U2 U_2 V2 V_2 \vdots UN1 U_{N-1} VN1 V_{N-1}

输出格式

答えを出力せよ。

题目大意

给你一个 nn 个顶点的无根树, 记 dd 为它的直径.

求使得顶点集合 SS 中任意两个顶点的距离为 dd 的集合个数, 对 998244353998244353 取模.

SS 中至少要有两个数.

n2×105n \le 2 \times 10^5.

5
1 2
1 3
1 4
4 5
2
4
1 2
1 3
1 4
4

提示

制約

  • 2  N  2× 105 2\ \leq\ N\ \leq\ 2\times\ 10^5
  • 1  Ui,Vi  N 1\ \leq\ U_i,V_i\ \leq\ N
  • Ui  Vi U_i\ \neq\ V_i
  • 入力は全て整数である。
  • 与えられるグラフは木である。

Sample Explanation 1

与えられた木は 5 5 頂点からなり、直径は 3 3 です。 2 2 頂点の組であって、その間の距離が 3 3 であるようなものは (2,5) (2,5) , (3,5) (3,5) しか存在しないため、 条件をみたす塗り方は { 2,5} \lbrace\ 2,5\rbrace { 3,5} \lbrace\ 3,5\rbrace 2 2 通りとなります。 { 2,3,5} \lbrace\ 2,3,5\rbrace は頂点 2 2 と頂点 3 3 の間の距離が 2 2 であるため条件をみたさないことに注意してください。

Sample Explanation 2

直径は 2 2 であり、条件をみたす塗り方は { 2,3} \lbrace\ 2,3\rbrace , { 2,4} \lbrace\ 2,4\rbrace , { 3,4} \lbrace\ 3,4\rbrace , { 2,3,4} \lbrace\ 2,3,4\rbrace 4 4 通りとなります。