计蒜客网络延时

问题描述

某计算机网络中存在 n 个路由，每个路由代表一个子网。路由之间有 n - 1 条互通关系，使得这 n 个网络之间任意两个网络都可以直接联通，或者通过其他网络间接连通。  为了测试组建的网路的性能，假设相邻的路由之间的数据传输需要一单位时间，现在需要知道任意两个路由之间传输数据最多需要多长时间。

输入格式  

第一行一个整数 n(2≤n≤50000) 表示网络中路由个数。接下来 n - 1 行，每行输入 u,v(1≤u,v≤n) ，表示路由 u, v 相连。  

输出格式  

输出一行表示答案。  

样例输入  

8 

6 3 

3 7 

3 4 

7 5 

5 1 

6 8 

5 2 

样例输出  

5

思路：

一道求树的直径的模板题。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=50010;
struct Way{
	int to,next;
} way[N<<1];
int n,a,b,cnt,mid,deep,head[N]; bool vis[N];
void add(int u,int v){
	way[++cnt].to=v;
	way[cnt].next=head[u];
	head[u]=cnt;
}
void dfs(int cur,int d){
	if(d>deep){
		mid=cur; deep=d;
	}
	for(int i=head[cur];i;i=way[i].next)
		if(!vis[way[i].to]){
			vis[way[i].to]=true;
			dfs(way[i].to,d+1);
		}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n-1;i++){
		scanf("%d %d",&a,&b);
		add(a,b); add(b,a);
	}
	deep=-1; vis[1]=true; dfs(1,0);
    memset(vis,0,sizeof(vis));
	deep=-1; vis[mid]=true; dfs(mid,0);
	printf("%d",deep);
	return 0;
}