E-受限观光树(sightseeing)

题目背景

在一座大型景区中,所有观光点通过步道连接起来,整体形成一棵树。

入口位于 11 号观光点。游客从入口出发,只能沿着步道不断向下前进,最终到达某个 终点观景点 (也就是没有继续分叉的叶子结点)。

但是景区中有些观光点过于拥挤。为了保证游玩体验,管理员规定:

在任意一条游览路径上,连续经过的拥挤观光点数量不能超过 mm

你的任务是统计:从入口出发,最终有多少个叶子结点可以作为合法终点。

题目描述

给定一棵包含 nn 个结点的树,结点编号为 11nn,并规定结点 11 为根。

对于每个结点 ii,给定一个值 aia_i

  • ai=1a_i=1 表示该观光点拥挤;
  • ai=0a_i=0 表示该观光点不拥挤。

如果从根结点 11 到某个叶子结点的简单路径上,任意位置都不存在长度大于 mm 的连续 1 段,那么这个叶子结点就是合法终点。

请你输出合法终点的数量。

输入格式

第一行输入两个整数 n,mn,m,分别表示树的结点数以及允许的连续拥挤点数量上限。

第二行输入 nn 个整数 a1,a2,,ana_1,a_2,\dots,a_n,其中 ai0,1a_i \in {0,1}

接下来输入 n1n-1 行,每行两个整数 u,vu,v,表示树上存在一条连接 uuvv 的无向边。

输出格式

输出一个整数,表示合法叶子结点的个数。

输入输出样例

输入 #1

7 1
1 1 0 0 1 0 1
1 2
1 3
2 4
2 5
3 6
3 7

输出 #1

2

样例解释

树的根为 11 号结点。

各条从根到叶子的路径分别为:

  1. 1241 \to 2 \to 4 路径上的拥挤情况为: 1 1 0\text{1 1 0} , 出现了连续 22 个拥挤点,而 m=1m=1,因此这条路径不合法。

  2. 1251 \to 2 \to 5 路径上的拥挤情况为: 1 1 1\text{1 1 1} , 出现了连续 33 个拥挤点,因此这条路径不合法。

  3. 1361 \to 3 \to 6 路径上的拥挤情况为: 1 0 0\text{1 0 0} , 连续拥挤点的最大数量为 11,合法。

  4. 1371 \to 3 \to 7 路径上的拥挤情况为: 1 0 1\text{1 0 1} , 连续拥挤点的最大数量仍为 11,合法。

因此合法叶子结点共有 22 个。

数据范围

对于所有测试数据,保证:$1 \le n \le 2 \times 10^5,\quad 0 \le m \le n,\quad a_i \in {0,1}$

并保证给出的边构成一棵树。

本题共设 2020 个测试点,各测试点满足的限制如下:

测试点编号 分值 nn mm 特殊性质
1\text{1} 55 10\le 10 =0=0 AA
2\text{2} n\le n BB
3\text{3} 20\le 20 CC
4\text{4} 无特殊性质
5\text{5} 100\le 100 =0=0 AA
6\text{6} n\le n DD
7\text{7} 1000\le 1000 EE
8\text{8} 无特殊性质
9\text{9} 2000\le 2000 5\le 5
10\text{10} 5000\le 5000 n\le n BB
11\text{11} 104\le 10^4 CC
12\text{12} 2×104\le 2 \times 10^4 10\le 10 无特殊性质
13\text{13} 5×104\le 5 \times 10^4 n\le n FF
14\text{14} 8×104\le 8 \times 10^4 无特殊性质
15\text{15} 105\le 10^5 DD
16\text{16} 1.2×105\le 1.2 \times 10^5 EE
17\text{17} 1.5×105\le 1.5 \times 10^5 无特殊性质
18\text{18} 1.8×105\le 1.8 \times 10^5 FF
19\text{19} 2×105\le 2 \times 10^5 =0=0=n=n AAGG
20\text{20} n\le n 无特殊性质

特殊性质 AA:保证 m=0m=0

特殊性质 BB:保证树是一条链。

特殊性质 CC:保证树是一棵星形树,即所有边都与结点 11 相连。

特殊性质 DD:保证所有结点的 ai=0a_i=0

特殊性质 EE:保证所有结点的 ai=1a_i=1

特殊性质 FF:保证除根结点外,每个结点的儿子数不超过 22

特殊性质 GG:保证 m=nm=n

相关

在下列比赛中:

「果壳杯」 ROUND 44 (Div. 4)