796C - Bank Hacking
阿新 • • 發佈:2018-12-09
題意:有N個銀行,每個銀行的的 xx值 為 a[i] , 銀行i被hack 需要的值 為 >=a[i] ,如果銀行i被hack,那麼 i的鄰接銀行j的 xx值加1,並且j的xx值加1。問需要最小的值,hack所有的銀行。
答案只可能 三種 m, m+1,m+2;
考慮 m + 2 , 如果考慮從i點hack, 辣麼i和i的鄰接 值 等於 m 的數量< m的所有數量的話, 則必然 有一個m的值要加2.
考慮 m+1 , 如果考慮從i點hack, 辣麼i的鄰接值出現m的話,則必有一個m+1. 或 m 的臨接點值等於m-1的數量 < m-1的所有數量,則必有一個m-1+2.
#include <bits\stdc++.h>
using namespace std;
const int maxn = 3*1e5+5;
#define INF -1e9;
vector<int> G[maxn];
int a[maxn],n;
int slove(){
int x = 0, y = 0;
int maxval = INF;
for(int i=1;i<=n;i++){
if(a[i]>maxval) maxval = a[i];
}
for(int i=1;i<=n;i++){
if (a[i]==maxval) x++;
if(a[i]==maxval-1) y++;
}
int xx = x, yy = y;
int flag1=1, flag2=1;
for(int i=1;i<=n;i++){
int flag3 = 1;
if(a[i]==maxval) x--;
else if(a[i]==maxval-1) y--;
for(int j=0;j<G[i].size();j++){
if(a[ G[i][j] ] == maxval){
x--;
flag3 = 0 ;
}
if(a[ G[i][j] ] == maxval-1) y--;
}
if(x==0){
flag1 = 0;
if( y==0&&flag3) flag2=0;
}
x = xx, y = yy;
}
if(flag1 == 0) {
if(flag2 == 0) return maxval;
else return maxval+1;
}
else return maxval+2;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<n;i++){
int x,y; scanf("%d%d",&x,&y);
G[x].push_back(y);
G[y].push_back(x);
}
int ans = slove();
printf("%d\n",ans);
return 0;
}