第十三届安徽省大学生程序设计大赛_D太空供水
2022-07-01 17:58 作者:Clayton_Zhou | 我要投稿
题目描述
空间站各舱室呈树状分布,一共有N个舱室,使用管道相连。经过统计得到了哪些太空舱现在有用水需求,目前需要给这些有需求的太空舱通水,但是初始只能在其中M个舱室安装水源。如果一个太空舱获得了水源,那么与它相连的太空舱可以花费1时间单位通过管道也获得水源。现在小明需要找出安装初始水源位置,使得在最短时间内,所有有用水需求的太空舱都可以获得水源。
输入说明
第一行是两个整数N,M。(1≤M≤N≤300000)
接下来一行有N个整数0和1,其中第i个数为1表示编号为i的舱室有用水需求。
接下来N-1行每行有两个数A,B,表示A和B之间有一条管道相连。
输出说明
一个整数, 表示使所有有用水需求的太空舱得到供水的最短时间。
输入样例
7 2
1 0 1 1 0 1 1
1 3
2 3
3 4
4 5
5 6
5 7
输出样例
1