题目链接: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;
}