CodeForces 796C Bank Hacking

题目链接：http://codeforces.com/contest/796/problem/C
题意：现有n个银行，要黑掉一个银行的话需要的能力为a[i]，有n-1条线将这n个银行连在一起，现有两种定义半相邻与相邻，i与j相连，j与k相连，则i与j半相邻，i与k为相邻，如果黑掉第i加银行，则要黑掉和他半相邻的银行j需要的能力为a[j]+1，要黑掉第二家银行必须和之前黑掉的银行相连，只有自身能力大过银行的，才可以黑掉他，现在让你黑掉所有银行，问你最小需要的能力是多少。
解析：首先如果随便找一家银行来黑的话，只有半相邻的要+1，其他则+2，画个图就可以发现这个规律，如果要黑掉所有银行，答案至少是max(a[1~n])，现在先判断几种情况：
+ 只有一个最大值
- 如果他相邻的银行的能力为max-1，如果是的话ans = max+1
- 否则ans = max
+ 不止一个最大值
- 假设i为最大值max，j也为最大值max，但是i和j不相邻，那么答案肯定是 ans = max+2
- 只有半相邻的为最大值的话，ans = max+1

写法是，开两个数组标记与最大值相连的点，和与最大值-1相连的点，然后记录最大值出现了多少次，最大值-1出现了多少次，然后按上述方法讨论即可

#include <iostream>#include <cstdio>#include <algorithm>#include <vector>#include <cstring>#include <queue>#include <cmath>#include <map>using namespace std;const int maxn = 1e6+100;const int inf = 0x7fffffff;const int mod = 1e9;int a[maxn];int vis[maxn];int c1[maxn];int c2[maxn];vector<int>G[maxn];int main(void){    int n;    scanf("%d",&n);    int ans = -inf;    int cnt1 = 0,cnt2 = 0;    for(int i=1;i<=n;i++)    {        scanf("%d",&a[i]);        ans = max(ans,a[i]);    }    for(int i=0;i<n-1;i++)    {        int x,y;        scanf("%d %d",&x,&y);        G[x].push_back(y);        G[y].push_back(x);    }    memset(c1,0,sizeof(c1));    memset(c2,0,sizeof(c2));    int tmp = -1;    for(int i=1;i<=n;i++)    {        if(ans==a[i])        {            c1[i]++;            for(unsigned j = 0;j<G[i].size();j++)                c1[G[i][j]]++;            cnt1++;            tmp = i;        }        if(ans==a[i]+1)        {            c2[i]++;            for(unsigned j = 0;j<G[i].size();j++)                c2[G[i][j]]++;            cnt2++;        }    }    int m1 = 0,m2 = 0;    for(int i=1;i<=n;i++)    {        m1 = max(m1,c1[i]);        m2 = max(m2,c2[i]);    }    if(cnt1==1 && c2[tmp]==cnt2)        printf("%d\n",ans);    else if(m1==cnt1)        printf("%d\n",ans+1);    else        printf("%d\n",ans+2);    return 0;}