Bank Hacking

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<string>
#include<stack>
#include<vector>
#include<set>
#include<map>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=1e5+10;
vector<int>v[N];
multiset<int>ms;//多重集合
int a[N];
int n,s,t;
int main(){


    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",a+i);
        ms.insert(a[i]);
    }
    for(int i=1;i<=n-1;i++){
        scanf("%d %d",&s,&t);
        v[s].push_back(t);
        v[t].push_back(s);
    }
    int ans=1<<30;
    multiset<int>::iterator it;
    for(int i=1;i<=n;i++){//遍历n家银行，找到需要最少能量的银行
        //存下银行i的值后，在ms中删除
        int x=a[i];
        ms.erase(ms.find(a[i]));
        //ms.erase(a[i]) 错误会删掉所有相同元素
        //找到跟i相邻的银行，比较后，在ms中删除
        for(int j=0;j<v[i].size();j++){
            int k=v[i][j];
            x=max(x,a[k]+1);
            ms.erase(ms.find(a[k]));
        }
        //比较距离大于2的银行能量最大值+2和x的大小，保证每个银行都可以攻击
        if(!ms.empty()){
            it=ms.end();
            x=max(x,*(--it)+2);
            //由于ms.end()指向最后一位,而不是最后一个元素所以--it;
            //不能 it--，会出错
        }
        //还原ms中删除的元素
        ms.insert(a[i]);
        for(int j=0;j<v[i].size();j++){
            int k=v[i][j];
            ms.insert(a[k]);
        }
        ans=min(x,ans);//找到所有情况中最少的能量
    }
    printf("%d\n",ans);
}