← 查看本卷题目 第 296 / 324 题

树上移动

1 秒 / 测试点 · 512 MiB · 20 个测试点。

题面描述

小杨有一棵包含 nn 个节点的树,其中节点的编号从 11 到 nn,每个节点的颜色要么是白色要么是黑色。小杨可以任意选择节点 ss 和节点 tt 并从节点 ss 出发移动到节点 tt,移动过程中小杨不能够经过重复节点。

小杨希望自己在至多经过 kk 个黑色节点的前提下,经过的总节点数尽可能多,请你帮小杨选择经过最多的节点数是

多少。

输入格式

第一行包含两个正整数 n,kn,k,代表节点数量和至多经过的黑色节点数。

第二行包含 nn 个正整数 a1,a2,…,ana_1,a_2,\ldots,a_n,代表节点颜色,如果 ai=0a_i=0,代表节点颜色为白色,如果 ai=1a_i=1,代表节点颜色为黑色。

之后 n−1n-1 行,每行包含两个正整数 ui,viu_i,v_i,代表存在一条连接节点 uiu_i 和 viv_i 的边。

输出格式

输出一个正整数,代表最多经过的节点数。

样例

5 10 0 1 1 11 22 32 51 4
3
子任务编号 数据点占比 nn kk 特殊性质
1 20% ≤100\le100 ≤100\le100 树的形态为一条链
2 20% ≤1000\le1000 00
3 60% ≤1000\le1000 ≤1000\le1000

对于全部数据,保证有 1≤n≤1000,0≤k≤1000,0≤ai≤11\le n\le1000,0\le k\le1000,0\le a_i\le1。