Codeforces 1294 F.Three Paths on a Tree (树的直径)

题目链接

题意：
给一颗树，让你找到三个不同的点，使得三个点之间的简单路径所覆盖的边集数量最大
输出最大覆盖到的边以及三个点的编号

解法：
容易看出树的直径中的两端点$(a,b)$必然是三个点其二，重点在于第三个点的选取。
假如现在选到的点为点$c$ ，则三个点覆盖到的边的数量 $=[dis(a,b)+dis(a,c)+dis(b,c)]/2$
(点c 到a / b的简单路径必然与直径(a,b)有交点)
所以我们只需要先求出树的直径两端点以及端点到其他点的距离，再枚举第三个点即可

Code:

const int MX =2e5 + 7;

vector<int>e[MX];
int dis[MX];
int bfs(int s){
	clr(dis,0);
	queue<int>q;q.push(s);
	int u;
	while(!q.empty()){
		u = q.front();q.pop();
		for(auto v:e[u]){
			if(!dis[v] && v != s){
				q.push(v);
				dis[v] = dis[u] + 1;
			} 
		}
	}
	return u;
}
int sum[MX];
int main()
{
	int n;scanf("%d",&n);
	for(int i = 1;i < n;++i){
		int u,v;scanf("%d %d",&u,&v);
		e[u].pb(v);e[v].pb(u);
	}
	int x = bfs(1), y = bfs(x), ans = 0, pos;
	for(int i = 1; i <= n;++i){
		sum[i] += dis[y] + dis[i];
	}
	bfs(y);
	for(int i = 1;i <= n;++i){
		sum[i] += dis[i];
		if(i != x && i != y && sum[i] > ans){
			pos = i;ans = sum[i];
		}
	}
	printf("%d\n%d %d %d\n", ans / 2,x,y,pos);
	return 0;
}