[BZOJ 2657][Zjoi2012]旅游(journey)：树的直径

/*
User:Small
Language:C++
Problem No.:2657
*/
#include<bits/stdc++.h>
#define ll long long
#define inf 999999999
using namespace std;
const int M=2e5+5,mod=5e6;
int n,tot,fir[M],dis[M];
struct edge{
int v,nex;
}e[M<<1];
void add(int u,int v){
e[++tot]=(edge){v,fir[u]};
fir[u]=tot;
}
struct Hash{
int fir[mod+5],tot,nex[M*3],v[M*3];
ll w[M*3];
void check(int a,int b,int x){
ll num=(ll)a*n+b;
int cc=num%mod;
for(int i=fir[cc];i;i=nex[i])
if(w[i]==num){
add(v[i],x);
add(x,v[i]);
return;
}
tot++;
nex[tot]=fir[cc];
w[tot]=num;
v[tot]=x;
fir[cc]=tot;
}
}h;
int bfs(int s){
memset(dis,127,sizeof(dis));
queue<int> q;
dis[s]=0;
q.push(s);
int res=0;
while(!q.empty()){
int u=q.front();
q.pop();
res=u;
for(int i=fir[u];i;i=e[i].nex){
int v=e[i].v;
if(dis[v]>dis[u]+1){
dis[v]=dis[u]+1;
q.push(v);
}
}
}
return res;
}
int main(){
freopen("data.in","r",stdin);//
scanf("%d",&n);
for(int i=1;i<=n-2;i++){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
if(b>c) swap(b,c);
if(a>c) swap(a,c);
if(a>b) swap(a,b);
h.check(a,b,i);
h.check(a,c,i);
h.check(b,c,i);
}
int u=bfs(1);
printf("%d\n",dis[bfs(u)]+1);
return 0;
}