Codeforces Round #364 (Div. 1)B. Connecting Universities

题目链接:传送门

题目大意:n个点构成一棵树，给定 k*2 点，要分成 k 组，使每组点之间的距离之和最大。

题目思路:因为是求距离之和最大，所以我们可以知道这样一个性质。如果以一条边为界，两边的子树均有给定的点，则这条边一定会经过 min（左边的给定点数，右边的给定点数）次。

　　　　 那么这条边的贡献就是经过的次数。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <cctype>
#include <queue>
#include <string>
#include <vector>
#include<functional>
#include <set>
#include <map>
#include <climits>
#define lson root<<1,l,mid
#define rson root<<1|1,mid+1,r
#define fi first
#define se second
#define ping(x,y) ((x-y)*(x-y))
#define mst(x,y) memset(x,y,sizeof(x))
#define mcp(x,y) memcpy(x,y,sizeof(y))
using namespace std;
#define gamma 0.5772156649015328606065120
#define MOD 1000000007
#define inf 0x3f3f3f3f
#define N 200005
#define maxn 10005
typedef pair<int,int> PII;
typedef long long LL;

int n,m,k,a[N];
LL ans;
vector<int>V[N];
void dfs(int x,int fa){
    for(int u:V[x]){
        if(u==fa)continue;
        dfs(u,x);
        a[x]+=a[u];
        ans+=min(a[u],k-a[u]);
    }
}
int main(){
    int i,j,group,x,y;
    scanf("%d%d",&n,&k);k<<=1;
    for(i=1;i<=k;++i){
        scanf("%d",&x);
        a[x]=1;
    }
    for(i=1;i<n;++i){
        scanf("%d%d",&x,&y);
        V[x].push_back(y);
        V[y].push_back(x);
    }
    dfs(1,-1);
    printf("%I64d\n",ans);
    return 0;
}