开始: 2025-07-04 17:00:00

暑假训练赛03专题:逆元,组合数,倍增算法

结束: 2025-07-04 21:00:00
当前  2025-07-17 20:00:30  类型: IOI  状态: 已经结束 

P4. 量子通信
描述

在未来的量子通信网络中,有 n 个量子节点(编号为 1n)通过量子纠缠通道连接形成一棵树状结构。

每个节点有两种状态:稳定态(用 0 表示)和干扰态(用 1 表示)。当信息在网络中传输时,必须遵循量子不可克隆原理,即每个节点只能被访问一次。

现在需要从任意节点 s 向节点 t 传输量子信息,同时要求路径上最多经过 k 个处于干扰态的节点。

请你设计一条传输路径,使得经过的节点总数最多,以最大化信息传输的稳定性。


输入

- 第一行包含两个正整数 nk,分别表示节点数量和至多允许经过的干扰态节点数。

- 第二行包含 n 个整数 a_1, a_2, \dots, a_n,其中 a_i = 0 表示节点 i 处于稳定态,a_i = 1 表示节点 i 处于干扰态。

- 接下来的 n-1 行,每行包含两个正整数 u_iv_i,表示节点 u_iv_i 之间存在一条量子纠缠通道。


输出

输出一个整数,表示满足条件的最长传输路径上的节点总数。

样例

输入

5 1
0 0 1 1 1
1 2
2 3
2 5
1 4

输出

3
提示

数据范围

对于所有数据,保证 1 \leq n \leq 10000 \leq k \leq 1000,且每个节点的状态为 01


提交

题目参数
时间限制 1 秒
内存限制 128 MB
提交