第十三屆安徽省大學(xué)生程序設(shè)計大賽_D太空供水
2022-07-01 17:58 作者:Clayton_Zhou | 我要投稿
題目描述
空間站各艙室呈樹狀分布,一共有N個艙室,使用管道相連。經(jīng)過統(tǒng)計得到了哪些太空艙現(xiàn)在有用水需求,目前需要給這些有需求的太空艙通水,但是初始只能在其中M個艙室安裝水源。如果一個太空艙獲得了水源,那么與它相連的太空艙可以花費1時間單位通過管道也獲得水源。現(xiàn)在小明需要找出安裝初始水源位置,使得在最短時間內(nèi),所有有用水需求的太空艙都可以獲得水源。
輸入說明
第一行是兩個整數(shù)N,M。(1≤M≤N≤300000)
接下來一行有N個整數(shù)0和1,其中第i個數(shù)為1表示編號為i的艙室有用水需求。
接下來N-1行每行有兩個數(shù)A,B,表示A和B之間有一條管道相連。
輸出說明
一個整數(shù), 表示使所有有用水需求的太空艙得到供水的最短時間。
輸入樣例
7 2
1 0 1 1 0 1 1
1 3
2 3
3 4
4 5
5 6
5 7
輸出樣例
1
標(biāo)簽: