TYVJ 4865 天天和树tree || 清北学堂金秋杯大奖赛

题目描述：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 110000
inline void read(int &x){
x=0; int f=1; char c=getchar();
while(c>'9'||c<'0'){ if(c=='-') f=-1; c=getchar(); }
while(c>='0'&&c<='9'){ x=x*10+c-'0'; c=getchar(); } x*=f;
}
struct Edge{ int to,next; }e[MAXN<<1];
int head[MAXN],fa[MAXN],dep[MAXN],pos,Max,num,n,tot;
bool vis[MAXN];

void dfs1(int u){
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(fa[u]==v) continue;
dep[v]=dep[u]+1,fa[v]=u;
if(dep[v]>Max) Max=dep[v],pos=v;
dfs1(v);
}
}

inline void Add_Edge(int u,int v){
e[++tot].to=v,e[tot].next=head[u],head[u]=tot;
}

bool dfs2(int u){
if(u==pos){ vis[u]=true;dep[u]=0;return vis[u]; }
bool flag=false;
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(v==fa[u]) continue;
fa[v]=u;
if(dfs2(v)==true) flag=true;
}
if(flag) dep[u]=0,vis[u]=true;
return flag;
}

void dfs3(int u){
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(v==fa[u]) continue;
fa[v]=u;
if(vis[v]) dep[v]=0;
else dep[v]=dep[u]+1;
Max=max(Max,dep[v]);dfs3(v);
}
}

int main(){
read(n);
for(int u,v,i=1;i<n;++i){
read(u),read(v);
Add_Edge(u,v),Add_Edge(v,u);
}
dep[1]=1,fa[1]=0,Max=-1;
dfs1(1);
memset(dep,0,sizeof dep );memset(fa,0,sizeof fa );
int pp=pos;
dfs1(pp);
bool tmp=dfs2(pp);
memset(fa,0,sizeof fa );
Max=-1;
dfs3(pos);
printf("%d",Max);
return 0;
}