Wannafly挑战赛1 C:MMSet2

题面

最优的点一定在给定点集中最远的两点的简单路径上。

#include
using namespace std;
const int maxn=3e5+10;
int head[maxn],To[maxn*2],Next[maxn*2],tol;
void add_edge(int x,int y)
{
    To[tol]=y;
    Next[tol]=head[x];
    head[x]=tol++;
}
int lg[maxn],f[maxn][22],dep[maxn];
void dfs(int v,int fa)
{
    dep[v]=dep[fa]+1;
    f[v][0]=fa;
    for(int i=1;(1<dep[y]) swap(x,y);
    while(dep[x]=0;i--)
    {
        if(f[x][i]!=f[y][i])
        {
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][0];
}
int s[1000005],len;
int get_dist(int st,int &d)
{
    int id=st;d=0;
    for(int i=1;i<=len;i++)
    {
        int ff=lca(st,s[i]);
        int dist=dep[st]+dep[s[i]]-2*dep[ff];
        if(dist>d)
        {
            d=dist;
            id=s[i];
        }
    }
    return id;
}
int main()
{
    for(int i=2;i