Codeforces 701E. Connecting Universities

题意：给了n个顶点，n-1条边的一棵树，给定一些顶点，需要分成k对，求这k对最大距离的和

分析：对于每一条边，求出以一个端点为根的子树含有的给定的点的个数x,则除这条边和这棵子树之外还有2*k-x个给定的点。因为每一个给定的点对应于唯一的另外一个顶点，而且要距离最大，所以一定是这条边两边的两个顶点且经过这条边的次数为min(x,2 * k - x)。DFS一遍求出所有边的贡献数即可

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 10;
bool vis[maxn];
ll cnt[maxn];
vector<int>vec[maxn];
ll ans;
int n,k;
void dfs(int now,int pre)
{
    if(vis[now])cnt[now] = 1;
    for(int i = 0; i < vec[now].size(); i++)
    {
        int v = vec[now][i];
        if(v == pre)continue;
        dfs(v,now);
        cnt[now] += cnt[v];
    }
    ans += min(2 * k - cnt[now],cnt[now]);
}
int main()
{

    cin>>n>>k;
    for(int i = 1; i <= 2 * k; i++)
    {
        int x;cin>>x;
        vis[x] = true;
    }
    for(int i = 1; i <= n - 1; i++)
    {
        int u,v;
        cin>>u>>v;
        vec[u].push_back(v);
        vec[v].push_back(u);
    }
    dfs(1,-1);
    cout<<ans<<endl;
    return 0;
}