atcoder#AGC008F. [AGC008F] Black Radius

[AGC008F] Black Radius

配点 : 19001900

問題文

NN 頂点の木があります。 頂点は 11 から NN まで番号が振られています。 各 1iN11 \leq i \leq N - 1 について、ii 番目の辺は頂点 aia_ibib_i を繋いでいます。 辺の長さはすべて 11 です。

いくつかの頂点はすぬけ君のお気に入りです。 お気に入りの頂点の情報は、長さ NN の文字列 ss として与えられます。 各 1iN1 \leq i \leq N について、頂点 ii がお気に入りならば sis_i1 で、頂点 ii がお気に入りでないならば sis_i0 です。

最初、頂点はすべて白色です。 すぬけ君は次の操作をちょうど 11 回だけ行います。

  • お気に入りの頂点 vv をひとつ選び、非負整数 dd をひとつ選ぶ。 頂点 vv から距離 dd 以内の頂点をすべて黒く塗る。

最終的な頂点の色の組合せとして考えられるものは何通りか求めてください。

制約

  • 2N2×1052 \leq N \leq 2 \times 10^5
  • 1ai,biN1 \leq a_i, b_i \leq N
  • グラフは木である。
  • s=N|s| = N
  • ss01 のみからなる。
  • ss には 1 が含まれる。

部分点

  • 13001300 点分のデータセットでは、ss1 のみからなる。

入力

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

NN

a1a_1 b1b_1

a2a_2 b2b_2

::

aN1a_{N - 1} bN1b_{N - 1}

ss

出力

最終的な頂点の色の組合せとして考えられるものは何通りか出力せよ。

4
1 2
1 3
1 4
1100
4

次の 44 通りです。

334d566ec1f4f38d23cd580044f1cd07.png

5
1 2
1 3
1 4
4 5
11111
11

このケースは部分点の制約を満たします。

6
1 2
1 3
1 4
2 5
2 6
100011
8