题意:
给一颗树,让你找到三个不同的点,使得三个点之间的简单路径所覆盖的边集数量最大
输出最大覆盖到的边以及三个点的编号
解法:
容易看出树的直径中的两端点
(
a
,
b
)
(a,b)
(a,b)必然是三个点其二,重点在于第三个点的选取。
假如现在选到的点为点
c
c
c ,则三个点覆盖到的边的数量
=
[
d
i
s
(
a
,
b
)
+
d
i
s
(
a
,
c
)
+
d
i
s
(
b
,
c
)
]
/
2
= [dis(a,b) + dis(a,c) + dis(b,c)] / 2
=[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;
}