题目描述
The Byteotian Cave is composed of n chambers and n-1 corridors that connect them. For every pair of chambers there is unique way to move from one of them to another without leaving the cave. Dynamite charges are set up in certain chambers. A fuse is laid along every corridor. In every chamber the fuses from the adjacent corridors meet at one point, and are further connected to the dynamite charge if there is one in the chamber. It takes exactly one unit of time for the fuse between two neighbouring chambers to burn, and the dynamite charge explodes in the instant that fire reaches the chamber it is inside.
We would like to light the fuses in some m chambers (at the joints of fuses) in such a way that all the dynamite charges explode in the shortest time possible since the fuses are lit. Write a program that will determine the minimum such time possible.
Byteotian Cave的结构是一棵N个节点的树，其中某些点上面已经安置了Bomb，现在需要点燃M个点上的引线引爆所有的Bomb。
某个点上的引线被点燃后的1单位时间内，在树上和它相邻的点的引线会被点燃。如果一个有Bomb的点的引信被点燃，那么这个点上的Bomb会爆炸。
求引爆所有Bomb的最短时间。
输入：
第一行是两个整数N,M。(1<=m<=n<=300000）
接下来一行有N个整数Di，第I个数为1表示该点有Bomb。
接下来N-1行每行有两个数A,B，表示A和B之间有一条边。
输出：
最短时间。
样例解释：
点燃3,5上的引线。
输入
The first line of the standard input holds two integers n and m (1<=M<=N<=300000)
, separated by a single space, that denote, respectively, the number of chambers in the cave and the number of chambers in which fire can be set to the fuses. The chambers are numbered from 1 to n . The next line contains n integers d1,d2…dn (Di属于{0,1}, separated by single spaces. If Di=1 , then there is dynamite in the -th chamber, and if di=0 , there is none. The following n -1 lines specify the corridors of the cave. Each of them holds two integers a,b (a<=a<b<=n), separated by a single space, denoting that there is a corridor connecting the chambers a and b . Every corridor appears exactly once in the description.
You may assume that in tests worth 10% of the points it holds additionally that n<= 10, while in tests worth 40% of the points it holds that N<=1000.
输出
The first and only line of the standard output should hold a single integer, equal to the minimum time it takes from lighting the fuses to the explosion of all the charges.
样例输入
7 2
1 0 1 1 0 1 1
1 3
2 3
3 4
4 5
5 6
5 7
样例输出
1
1 #pragma GCC optimize("O3")
2 #include <iostream>
3 #include <cstring>
4 #include <algorithm>
5 #include <cstdio>
6 using namespace std;
7 #define N 300005
8 int read() {
9 int s=0,f=1;
10 char ch=getchar();
11 for( ; ch<‘0‘||ch>‘9‘; f=(ch==‘-‘)?-1:f,ch=getchar()) ;
12 for( ; ch>=‘0‘&&ch<=‘9‘; s=s*10+(ch^48),ch=getchar()) ;
13 return s*f;
14 }
15 int n,m,tot,r[N],deep[N],Bomb[N],L,R,Ans,son[N],num;
16 struct EDGE{int to,next;} c[N<<1];
17 void add(int x,int y) {
18 c[++tot]=(EDGE){y,r[x]};
19 r[x]=tot;
20 }
21 int Judge[N],f[N];
22 //可以再往上传的长度 需要被炸得最长长度
23 void DFS(int u,int fa,int now) {
24 int mx1=-0x5f5f5f5f,mx2=-0x5f5f5f5f;
25 if(Bomb[u]) mx1=0;
26 for(int i=r[u]; ~i; i=c[i].next) {
27 if(c[i].to!=fa) {
28 DFS(c[i].to,u,now);
29 if(Judge[c[i].to]==1) mx1=max(mx1,f[c[i].to]+1);
30 if(Judge[c[i].to]==2) mx2=max(mx2,f[c[i].to]-1);
31 if(Judge[c[i].to]==-1||num>m) {Judge[u]=-1; return ;}
32 }
33 }
34 if(mx1>mx2) {
35 if(mx1>=now) {
36 ++num,Judge[u]=2,f[u]=now;
37 if(num>m) Judge[u]=-1;
38 } else Judge[u]=1,f[u]=mx1;
39 } else Judge[u]=2,f[u]=mx2;
40 }
41 bool check(int now) {
42 num=0;
43 DFS(1,1,now);
44 if(Judge[1]==-1) return false;
45 if(Judge[1]==1&&f[1]>=0) ++num;
46 return num<=m;
47 }
48 int main() {
49 R=n=read(),m=read();
50 memset(r,0xff,sizeof(r));
51 for(int x,i=1; i<=n; ++i) Bomb[i]=read();
52 for(int x,y,i=1; i<n; ++i) x=read(),y=read(),add(x,y),add(y,x);
53 while(L<=R) {
54 int mid=(L+R)>>1;
55 if(check(mid))Ans=mid,R=mid-1;
56 else L=mid+1;
57 } printf("%d",Ans);
58 return 0;
59 }
