1. 程式人生 > >796C - Bank Hacking

796C - Bank Hacking

題意:有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; }