题意:相当于若干个银行构成了一个无根树,我们需要把所有的银行hack掉,刚开始可以随意选一个,后来就只能选被hack过的邻居才行,每个银行都有他们的strength,hack它需要不比它小的strength值,每次hack一个银行后,它的邻居和邻居的邻居strength值都会+1,求hack全部的银行需要的最小strength值。
可以想,如果只有一个最大值,那么存在两种情况,如果所有银行的strength值-1的值都连在了这个最大值上,那么所需最大strength值就是这个最大值,反之所需最大strength值为这个最大值+1。
如果有多个最大值,也存在两种情况,如果存在一个点使得所有最大值到该点的距离<=1,那么所需strength值为这些银行最大值+1,反之所需strength值为最大值+2。
#include<bits/stdc++.h>
using namespace std;
int n,a[300005],ma=-1e9,man=0,ma_n=0,u,v,x;
vector<int>G[300005];
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
ma=max(ma,a[i]);
}
for(int i=1;i<=n;i++)
{
if(a[i]==ma)man ++;
if(a[i]==ma-1)ma_n++;
}
for(int i=0;i<n-1;i++)
{
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
if(man==1)
{
int k,sum=0;
for(k=1;k<=n;k++)
{
if(a[k]==ma)break;
}
for(int i=0;i<G[k].size();i++)
{
if(a[G[k][i]]==ma-1)sum++;
}
if(sum==ma_n)
{
cout<<ma<<endl;
return 0;
}
cout<<ma+1;
return 0;
}
else
{
int k,sum=0;
for(k=1;k<=n;k++)
{
if(a[k]==ma)break;
}
for(int i=0;i<G[k].size();i++)
{
if(a[G[k][i]]==ma)sum++;
}
if(sum==man-1)
{
cout<<ma+1;
return 0;
}
for(int i=0;i<G[k].size();i++)
{
sum=0;x=G[k][i];
if(a[x]==ma)sum++;
for(int j=0;j<G[x].size();j++)
{
if(a[G[x][j]]==ma)sum++;
}
if(sum==man)
{
cout<<ma+1;
return 0;
}
}
}
cout<<ma+2;
return 0;
}