codeforces 1294F Three Paths on a Tree 树的直径+贪心

https://vjudge.net/problem/CodeForces-1294F

题目大意：给一棵树，让你找到三个顶点$a、b、c$，使得$a-b、a-c、b-c$的路径所覆盖的变数尽可能大(同一条边被多次覆盖也只算$1$次)。

思路：贪心的思考一下，必定有一组最优解，使得$a、b$是树直径的两个端点。那么通过两次$dfs/bfs$求得树的直径后，枚举第三个端点即可得到最优解。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const int maxn=2e5+5;

struct Edge
{
    int to,nxt;
}edge[maxn<<1];

int n,tot;
int head[maxn],dis[maxn],disa[maxn],disb[maxn];
bool vis[maxn];

inline void addedge(int u,int v)
{
    edge[++tot].to=v,edge[tot].nxt=head[u],head[u]=tot;
}

int bfs(int s)
{
    memset(vis,0,sizeof(vis));
    queue<int> q;
    q.push(s);
    vis[s]=1,dis[s]=0;
    int u,v,res=s;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(int i=head[u];i;i=edge[i].nxt)
        {
            v=edge[i].to;
            if(!vis[v])
            {
                vis[v]=1,dis[v]=dis[u]+1;
                q.push(v);
                if(dis[v]>dis[res])
                    res=v;
            }
        }
    }
    return res;
}

int main()
{
    scanf("%d",&n);
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        addedge(u,v),addedge(v,u);
    }
    int a,b,c=0;
    a=bfs(1);
    b=bfs(a);
    memcpy(disa,dis,sizeof(dis));
    bfs(b);
    memcpy(disb,dis,sizeof(dis));
    for(int i=1;i<=n;i++)
    {
        if(i!=a&&i!=b&&disa[i]+disb[i]>disa[c]+disb[c])
            c=i;
    }
    int ans=(disa[c]+disb[c]+disa[b])/2;
    printf("%d\n%d %d %d\n",ans,a,b,c);
    return 0;
}