Codeforces Round #364 (Div. 2) E Connecting Universities

n个城市，2k个大学，两两配对，怎么连接才能使总距离最大。
看题后首先想到先找重心，然后所有大学向重心连接的距离和就是结果。后来想想不对，万一有超过k个大学在同一个子树里，那有些就不能向重心连接了。所以呢，就不求重心了，求一个这样的点，这个点所有的子树里的大学个数<=k，然后所有大学到这个点的距离和就是结果了。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN = 400100;
struct Edge
{
    int to,next;
};
Edge edge[MAXN];
int head[MAXN],tot;
int son[MAXN];
int dis[MAXN];
int pt[MAXN];
int n,k,root=1;

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

void dfs1(int u, int fa)
{
    if(dis[u]) son[u] = 1;
    int v;
    bool flag = true;
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        v = edge[i].to;
        if(v == fa) continue;
        dfs1(v,u);
        son[u] += son[v];
        if(son[v] > k) flag = false;
    }
    if(2*k-son[u] > k) flag = false;

    if(flag) root = u;
}

void dfs2(int u, int fa, int len)
{
    dis[u] = len;
    int v;
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        v = edge[i].to;
        if(v == fa) continue;
        dfs2(v,u,len+1);
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin >> n >> k;
    int u,v;
    for(int i = 0; i < 2*k; ++i)
    {
        cin >> pt[i];
        dis[pt[i]] = 1;
    }
    memset(head,-1,sizeof(head));
    tot = 0;
    for(int i = 0; i < n-1; ++i)
    {
        cin >> u >> v;
        addedge(u,v);
        addedge(v,u);
    }
    dfs1(1,-1);
    dfs2(root,-1,0);
    LL res = 0;
    for(int i = 0; i < 2*k; ++i)
        res += dis[pt[i]];
    cout << res << endl;
    return 0;
}