codeforce 796C Bank Hacking 智力题orz

题目:

There are n banks, numbered from 1 to n. There are also n - 1 wires connecting the banks. All banks are initially online. Each bank also has its initial strength: bank i has initial strength ai.

Let us define some keywords before we proceed. Bank i and bank j are neighboring if and only if there exists a wire directly connecting them. Bank i and bank j are semi-neighboring if and only if there exists an online bank k such that bank i and bank k are neighboring and bank k and bank j are neighboring.

When a bank is hacked, it becomes offline (and no longer online), and other banks that are neighboring or semi-neighboringto it have their strengths increased by 1.

To start his plan, Inzane will choose a bank to hack first. Indeed, the strength of such bank must not exceed the strength of his computer. After this, he will repeatedly choose some bank to hack next until all the banks are hacked, but he can continue to hack bank x if and only if all these conditions are met:

  1. Bank x is online. That is, bank x is not hacked yet.
  2. Bank x is neighboring to some offline bank.
  3. The strength of bank x is less than or equal to the strength of Inzane's computer.

Determine the minimum strength of the computer Inzane needs to hack all the banks.

Input

The first line contains one integer n (1 ≤ n ≤ 3·105) — the total number of banks.

The second line contains n integers a1, a2, ..., an ( - 109 ≤ ai ≤ 109) — the strengths of the banks.

Each of the next n - 1 lines contains two integers ui and vi (1 ≤ ui, vi ≤ nui ≠ vi) — meaning that there is a wire directly connecting banks ui and vi.

It is guaranteed that the wires connect the banks in such a way that Inzane can somehow hack all the banks using a computer with appropriate strength.

Output

Print one integer — the minimum strength of the computer Inzane needs to accomplish the goal.

Examples
input
5
1 2 3 4 5
1 2
2 3
3 4
4 5
output
5
input
7
38 -29 87 93 39 28 -55
1 2
2 5
3 2
2 4
1 7
7 6
output
93
input
5
1 2 7 6 7
1 5
5 3
3 4
2 4
output
8
Note

In the first sample, Inzane can hack all banks using a computer with strength 5. Here is how:

  • Initially, strengths of the banks are [1, 2, 3, 4, 5].
  • He hacks bank 5, then strengths of the banks become [1, 2, 4, 5,  - ].
  • He hacks bank 4, then strengths of the banks become [1, 3, 5,  - ,  - ].
  • He hacks bank 3, then strengths of the banks become [2, 4,  - ,  - ,  - ].
  • He hacks bank 2, then strengths of the banks become [3,  - ,  - ,  - ,  - ].
  • He completes his goal by hacking bank 1.

In the second sample, Inzane can hack banks 423157, and 6, in this order. This way, he can hack all banks using a computer with strength 93.


答案tutotial已经说得很清楚了,关键是意识到,如果v是被hack的点,hack完整个树后,那么与它(v)直接相邻的点的strenth+1,不直接相邻的点strenth+2.然后就是分别考虑有多个最大值和单个最大值。

case1:单个最大值。那么就考虑,1,次小值是不是为max-1,如果是 再考虑是不是所有次小值都连在了max上,如果是,res=max。否则,res=max+1.2,如果次小值不是max-1,res=max.

case2: 多个最大值。那么就考虑,1,所有最大值点是否连在一个点上(可能这个点本身就是最大值点,也可能不是)如果是,res=max+1,否则res=max+2.

code:

哎,挫代码,还是不要参考了。

#include<cstdio>
#include<map>
#include<cstring>
#define max(a,b) (a>b?a:b)
using namespace std;
typedef long long LL;
const int MAXN=3e5+5;
const LL INF=1e10+5;
long long  a[MAXN],head[MAXN],w[MAXN];
struct edge{int v,next;};
struct edge es[MAXN*2];
int tot;
map<LL ,int>cnt;
void init(){
    tot=0;
    memset(head,-1,sizeof(head));
}
void addEdge(int a,int b){
    es[tot].v=b;
    es[tot].next=head[a];
    head[a]=tot++;
}
bool flag;
LL big0,big2;
void dfs0(int x){
    int maxCnt=0;
    if(a[x]==big0)++maxCnt;
    for(int i=head[x];i!=-1;i=es[i].next){
        int to=es[i].v;
        if(a[to]==big0)++maxCnt;
    }
    if(!flag){
        flag=(maxCnt==cnt[big0]);
    }
}
void dfs2(int x){
    int big2Cnt=0;
    for(int i=head[x];i!=-1;i=es[i].next){
        int to=es[i].v;
        if(a[to]==big2)++big2Cnt;
    }
    if(!flag){
        flag=(big2Cnt==cnt[big2]);
    }
}
int main(void){
    int n;scanf("%d",&n);
    big0=-INF,big2=-INF;
    for(int i=1;i<=n;++i){
        scanf("%I64d",a+i);
        cnt[a[i]]++;
        if(a[i]>=big0){
            big2=max(big2,big0);
            big0=a[i];
        }else if(a[i]>big2)big2=a[i];
    }
    //printf("big0=%I64d,big2=%I64d\n",big0,big2);
    init();
    for(int i=1;i<n;++i){
        int a,b;scanf("%d%d",&a,&b);
        addEdge(a,b);
        addEdge(b,a);
    }
    flag=false;
    if(cnt[big0]>1){
        for(int i=1;i<=n;i++){
            dfs0(i);
        }
        printf("%I64d\n",flag?big0+1:big0+2);
    }
    else{
        if(big0-big2==1){
            for(int i=1;i<=n;++i){
                if(a[i]==big0)dfs2(i);
            }
            printf("%I64d\n",flag?big0:big0+1);
        }
        else{
            printf("%I64d\n",big0);
        }
    }
}


另外就是注意初始化的问题,例如数组大小,以及big0,big1的初始化。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值