#agc008f. [agc008_f]Black Radius
[agc008_f]Black Radius
题目描述
有一棵 个顶点的树。顶点从 到 编号。对于每个 ,第 条边连接顶点 和 。所有边的长度为 。
Snuke 喜欢其中一些顶点。他喜欢的顶点的信息以一个长度为 的字符串 给出。对于每个 ,若 Snuke 喜欢顶点 ,则 为 1
,否则为 0
。
初始时,所有顶点都是白色的。Snuke 将执行以下操作一次:
- 选择一个他喜欢的顶点 ,和一个非负整数 。然后,将所有距离 不超过 的顶点涂成黑色。
找出操作后可能的顶点颜色组合的数量。
约束条件
- 给定的图是一棵树。
- 由
0
和1
组成。 - 中至少包含一个
1
。
分数说明
- 在价值 分的测试集中, 仅包含
1
。
输入
输入以以下格式从标准输入给出:
输出
打印操作后可能的顶点颜色组合的数量。
示例 1
4
1 2
1 3
1 4
1100
输出 1
4
有以下四种可能的顶点颜色组合:
示例 2
5
1 2
1 3
1 4
4 5
11111
输出 2
11
此示例满足部分得分的额外约束条件。
示例 3
6
1 2
1 3
1 4
2 5
2 6
100011
输出 3
8