AtCoder Grand Contest 008
F
Black Radius
AtCoder
⌀ 3995
原题
← E→
翻译中…

Problem Statement

翻译中…

There is a tree with $N$ vertices. The vertices are numbered $1$ through $N$. For each $1 ≤ i ≤ N - 1$, the $i$-th edge connects vertices $a_i$ and $b_i$. The lengths of all the edges are $1$.

翻译中…

Snuke likes some of the vertices. The information on his favorite vertices are given to you as a string $s$ of length $N$. For each $1 ≤ i ≤ N$, $s_i$ is 1 if Snuke likes vertex $i$, and 0 if he does not like vertex $i$.

翻译中…

Initially, all the vertices are white. Snuke will perform the following operation exactly once:

翻译中…
  • Select a vertex $v$ that he likes, and a non-negative integer $d$. Then, paint all the vertices black whose distances from $v$ are at most $d$.
翻译中…

Find the number of the possible combinations of colors of the vertices after the operation.

翻译中…

Constraints

翻译中…
  • $2 ≤ N ≤ 2×10^5$
  • $1 ≤ a_i, b_i ≤ N$
  • The given graph is a tree.
  • $|s| = N$
  • $s$ consists of 0 and 1.
  • $s$ contains at least one occurrence of 1.
翻译中…

Partial Score

翻译中…
  • In the test set worth $1300$ points, $s$ consists only of 1.
翻译中…

Input

翻译中…

The input is given from Standard Input in the following format:

翻译中…

N
a_1 b_1
a_2 b_2
:
a_{N - 1} b_{N - 1}
s

翻译中…

Output

翻译中…

Print the number of the possible combinations of colors of the vertices after the operation.

翻译中…

样例

Input #1
4
1 2
1 3
1 4
1100
Output #1
4
The following four combinations of colors of the vertices are possible: ![334d566ec1f4f38d23cd580044f1cd07.png](https://atcoder.jp/img/agc008/334d566ec1f4f38d23cd580044f1cd07.png)
Input #2
5
1 2
1 3
1 4
4 5
11111
Output #2
11
This case satisfies the additional constraint for the partial score.
Input #3
6
1 2
1 3
1 4
2 5
2 6
100011
Output #3
8
题目信息
题号
agc008_f
时间限制
2 s
内存限制
256 MB

题面与样例版权归 AtCoder 及出题人所有;中文翻译由 AI 生成(CI 管线),如有出入请以原文为准。