codeforces round#408(div.2) C. Bank Hacking

题意：相当于若干个银行构成了一个无根树，我们需要把所有的银行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;
}