AT_joisc2008_flu インフルエンザ (Flu)

题目描述

JOI 市里住着 $N$ 个居民,他们被编号为 $1, 2, \ldots, N$。每个居民的健康状态用一个整数表示:健康状态为正数时,表示该居民健康;为负数时,则表示该居民感染了流感。 每天,流感会通过朋友关系传播。具体来说,如果居民 $i$ 感染流感,即健康状态为负,并且他与居民 $j$ 是朋友,那么居民 $j$ 的健康状态会减少 $1$。一旦健康状态低于或等于 $0$,居民 $j$ 就会被感染流感。 你的任务是计算在第 $D$ 天结束时,患上流感的居民总人数。

输入格式

输入通过标准输入提供。第一行包含两个整数 $N$ 和 $D$,分别表示居民的数量及观察的天数。 接下来的第二行包含 $N$ 个整数 $H_1, H_2, \ldots, H_N$,表示每个居民初始的健康状态。 随后有 $N-1$ 行,每行包括两个整数 $A_i$ 和 $B_i$,表示居民 $A_i$ 与居民 $B_i$ 是朋友。保证这些朋友关系形成了一棵树结构。

输出格式

输出一个整数,表示在第 $D$ 天结束时患有流感的居民数量。

说明/提示

- $2 \leq N \leq 10^5$ - $1 \leq D \leq 10^5$ - $-10^9 \leq H_i \leq 10^9$ - $1 \leq A_i, B_i \leq N$ - $A_i \neq B_i$ 上述数据范围保证朋友关系形成的是一棵树。 **本翻译由 AI 自动生成**