codeforce 796C Bank Hacking 智力题orz

来源:互联网 发布:java多态例子 编辑:程序博客网 时间:2024/05/23 23:12

题目:

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
51 2 3 4 51 22 33 44 5
output
5
input
738 -29 87 93 39 28 -551 22 53 22 41 77 6
output
93
input
51 2 7 6 71 55 33 42 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
原创粉丝点击