#2689. Sam的素描作品

Sam的素描作品

题目背景

【备注】 二分答案做法复杂度为 O(n2logn)O(n^2logn)。数据可加强至 n,m2000n,m\le2000,卡掉 O(n3)O(n^3)

题目描述

众所周知,SamSam 很喜欢画关于树的画,最近不是很空嘛(为什么很空?参见 SamSam 的春秋大梦)于是他反手就画出了一棵有 nn 个节点根为 11 的树。

然后他随意的把这 nn 个节点随机染成黑色或者白色。

现在他想知道,在这棵树上任意选择恰好 mm 个黑色节点,这 mm 个黑色节点之间的距离的最大值最小是多少?

输入格式

第一行输入两个整数 n,mn,m,,分别代表节点的个数以及你需要选取的黑色节点的个数。

接下来的一行输入 nn 个整数 a1,a2ana_1,a_2 \dots a_n 。 如果 ai=1a_i = 1,那么说明第 ii 个节点是黑色的,否则则是白色的。

接下来 n1n - 1 行,每行包含两个整数 x,yx,y 表示 x,yx,y 之间存在一条双向边。

输出格式

输出一行,代表答案。

输入输出样例 #1

输入 #1

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

输出 #1

2

说明/提示

样例解释

在第一组样例中,选择节点 1,2,41,2,4 或者 1,5,61,5,6,他们的距离最大值是 22

数据范围

测试点 n,mn,m 特殊性质
25%25\% 1mn101 \le m \le n \le 10
50%50\% 1mn801 \le m \le n \le 80
100%100\% 1mn1001 \le m \le n \le 100

特别的保证:输入的数据保证黑色节点至少有 mm 个,并且输入的数据一定是一棵树。