在未来的量子通信网络中,有 n 个量子节点(编号为 1 到 n)通过量子纠缠通道连接形成一棵树状结构。
每个节点有两种状态:稳定态(用 0 表示)和干扰态(用 1 表示)。当信息在网络中传输时,必须遵循量子不可克隆原理,即每个节点只能被访问一次。
现在需要从任意节点 s 向节点 t 传输量子信息,同时要求路径上最多经过 k 个处于干扰态的节点。
请你设计一条传输路径,使得经过的节点总数最多,以最大化信息传输的稳定性。
- 第一行包含两个正整数 n 和 k,分别表示节点数量和至多允许经过的干扰态节点数。
- 第二行包含 n 个整数 a_1, a_2, \dots, a_n,其中 a_i = 0 表示节点 i 处于稳定态,a_i = 1 表示节点 i 处于干扰态。
- 接下来的 n-1 行,每行包含两个正整数 u_i 和 v_i,表示节点 u_i 和 v_i 之间存在一条量子纠缠通道。
输出一个整数,表示满足条件的最长传输路径上的节点总数。
5 1 0 0 1 1 1 1 2 2 3 2 5 1 4
3
数据范围
对于所有数据,保证 1 \leq n \leq 1000,0 \leq k \leq 1000,且每个节点的状态为 0 或 1。
时间限制 | 1 秒 |
内存限制 | 128 MB |