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 自动生成**