cf796c 树形，思维题

一开始以为是个树形dp，特地去学了。。结果是个思维题

/*
树结构，设最大点权值为Max,则答案必在在区间[Max,Max+2]
证明ans <= Max+2
任取一个点作为根节点，那么去掉这个点之后其儿子结点，孙子结点的权值+1，同理，每去掉一个点之后其儿子结点，孙子结点的权值都会加上1
即每个点的权值最多只能被+2，即它的父亲，爷爷结点会导致其增加权值
考虑什么时候ans=Max:只有一个Max结点，以其为根，若有权值Max-1的点，那么这些点只有一个父亲就是Max
ans=Max+1:所有的Max有相同的父亲
怎么判断
*/
#include<bits/stdc++.h>
using namespace std;
#define maxn 300005

struct Edge{
    int to,next;
}edge[maxn<<];
int head[maxn],tot,n,a[maxn];

void init(){
    memset(head,-,sizeof head);
    tot=;
}
void addedge(int u,int v){
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}

int main(){
    int Max=-,tot1=,tot2=,u,v,x;
    init();
    scanf("%d",&n);
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
        Max=max(Max,a[i]);
    }
    for(int i=;i<n;i++){
        scanf("%d%d",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    for(int i=;i<=n;i++)
        if(Max==a[i]) tot1++,u=i;
        else if(Max-==a[i]) tot2++;

    int tmp=;
    if(tot1==){//只有一个Max
        for(int i=head[u];i!=-;i=edge[i].next){
             int v=edge[i].to;
            if(a[v]==Max-)tmp++;
        }
        if(tmp==tot2)//所有Max-1都是其子节点
             printf("%d",Max);
        else printf("%d",Max+);
        return ;
    }

    int vis[maxn]={},flag=;
    queue<int>q;
    q.push();vis[]=;
    while(!q.empty()){
        int u=q.front();q.pop();vis[u]=;
        int tmp=a[u]==Max?:;
        for(int i=head[u];i!=-;i=edge[i].next){
            int v=edge[i].to;
            if(!vis[v]) q.push(v);
            if(a[v]==Max)tmp++;
        }
        if(tmp==tot1){
            flag=;break;
        }
    }
    if(flag) printf("%d",Max+);
    else printf("%d",Max+);
}