問題一覧 > 通常問題

No.2892 Lime and Karin

レベル : / 実行時間制限 : 1ケース 8.000秒 / メモリ制限 : 512 MB / 標準ジャッジ問題
タグ : / 解いたユーザー数 32
作問者 : 👑 seekworser / テスター : 👑 AngrySadEight kemuniku
0 ProblemId : 11152 / 出題時の順位表 / 自分の提出
問題文最終更新日: 2025-01-30 10:44:59

注意

この問題の TL は厳しめに設定されています。 高速な言語の使用を推奨します。

問題文

頂点に11 から NN の番号が付けられた NN 頂点の木が与えられます。 i=1,2,,N1i=1,2,\dots,N-1 について ii 番目の辺は頂点 uiu_iviv_i を結びます。

木の各頂点には、ライムの木かカリンの木のいずれかが生えています。 文字列 SS が与えられ、 SSii 番目の文字が 1 のとき頂点 ii にはライムの木が、 SSii 番目の文字が 0 のとき頂点 ii にはカリンの木が生えています。

1lrN1 \le l \le r \le N を満たす頂点の組 (l,r)(l, r) であって、 ll から rr への両端を含む単純パス上の頂点に生えているライムの木の本数が カリンの木の本数よりも多いものの個数を答えてください。

単純パスとは? 「単純パス」とは頂点列 v1,v2,,vkv_1, v_2, \dots, v_k であって、 1ik11 \le i \le k-1 について viv_ivi+1v_{i+1} が辺によって直接結ばれており、 かつ全ての 1i<jk1 \le i < j \le k について viv_ivjv_j が異なるものを指します。 木において、両端を指定した単純パスは一意に定まります。

入力

NN
u1u_1 v1v_1
u2u_2 v2v_2
\vdots
uN1u_{N-1} vN1v_{N-1}
SS

入力

  • 1N1000001 \le N \le 100000
  • 1ui<viN(1iN1)1 \le u_i < v_i \le N \quad (1 \le i \le N-1)
  • N,ui,viN, u_i, v_i は全て整数 (1iN1)(1 \le i \le N-1)
  • S=N|S| = N
  • SS0 または 1 の文字からなる。

出力

答えを 11 行で出力せよ。

サンプル

サンプル1
入力
5
1 2
1 3
3 4
3 5
11001
出力
7

(l,r)=(1,1),(1,2),(1,5),(2,2),(2,3),(2,5),(5,5)(l, r) = (1, 1), (1, 2), (1, 5), (2, 2), (2, 3), (2, 5), (5, 5) が条件を満たします。

サンプル2
入力
1
1
出力
1
サンプル3
入力
10
2 8
4 9
1 6
3 7
1 8
5 8
1 10
1 7
7 9
0111110100
出力
16

提出するには、Twitter 、GitHub、 Googleもしくは右上の雲マークをクリックしてアカウントを作成してください。