标签：maxx int CF796C setIO Hacking define d2 Bank d1

思路十分简单，答案只有 3 种可能，但是有一些细节需要额外注意一下. 

code: 

#include <bits/stdc++.h>
#define N 300002    
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std; 
int val[N],hd[N],to[N<<1],nex[N<<1],d1[N],d2[N],n,edges,maxx,mx,m2,cnt,uu;             
void add(int u,int v) 
{
    nex[++edges]=hd[u],hd[u]=edges,to[edges]=v; 
}    
void dfs(int u,int ff) 
{
    if(val[u]==mx) d1[u]=0, uu=u;         
    for(int i=hd[u];i;i=nex[i]) 
    {
        int v=to[i]; 
        if(v==ff) continue;   
        dfs(v,u);    
        if(d1[v]+1>d1[u]) 
        {
            d2[u]=d1[u],d1[u]=d1[v]+1; 
        }
        else if(d1[v]+1>d2[u]) d2[u]=d1[v]+1;    
    }
    maxx=max(d1[u]+d2[u], maxx); 
}
int main() 
{ 
    int i,j; 
    // setIO("input");     
    mx=-1000300000; 
    m2=mx;   
    scanf("%d",&n);    
    for(i=1;i<=n;++i) 
    {
        scanf("%d",&val[i]),mx=max(mx,val[i]);   
    }
    for(i=1;i<=n;++i) if(val[i]<mx) m2=max(m2, val[i]); 
    for(i=1;i<=n;++i) if(val[i]==m2) ++cnt;    
    for(i=1;i<n;++i) 
    {
        int u,v;      
        scanf("%d%d",&u,&v),add(u,v),add(v,u); 
    }   
    memset(d1,-0x3f,sizeof(d1));         
    memset(d2,-0x3f,sizeof(d2));             
    dfs(1,0);         
    if(maxx==0) 
    { 
        if(m2!=mx-1) 
            printf("%d\n",mx);          
        else 
        {   
            for(int i=hd[uu];i;i=nex[i])
            {
                int v=to[i]; 
                if(val[v]==m2) --cnt;  
            }
            if(cnt) printf("%d\n",mx+1); 
            else printf("%d\n",mx);   
        }
    }
    else if(maxx<=2) printf("%d\n",mx+1);                        
    else printf("%d\n",mx+2); 
    return 0; 
}

标签：maxx,int,CF796C,setIO,Hacking,define,d2,Bank,d1
来源： https://www.cnblogs.com/guangheli/p/11619272.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。