Codeforces700B Connecting Universities(算贡献)

http://codeforces.com/problemset/problem/700/B
题意：n个点m条边，现在告诉你2k个点，让你分成k组，每组两点之间距离的总和最大。
题解：首先一下子就能发现肯定是这个点连离他最远的，这样贪心，于是先入为主了，就想办法想找重心，把他们分成两堆，然后重构这棵树，但是，有可能会找不到平分两堆的方法，所以GG。但是这个想法也没有错误，其实就是这样，但是应该找一种，对于两个不一样大小的堆也能计算的方法，那就是算每条边对答案的贡献，如果对于这条边，左边有a个标记点，右边有2k-a个标记点，那么对于上面的贪心来说，肯定是左边都往右边连，那么这条边贡献就是$min(a,2k-a)$，然后就是绝世傻题了。

唉这个套路做了这么久，还是不能熟练使用啊GG,这题做不出需要反思

---

代码：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           200005
#define   MAXN          1000005
#define   maxnode       15
#define   sigma_size    30
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-4;
const LL    mod    = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
 }
/*****************************************************/

int vis[MAX];
vector<int> v[MAX];
int num[MAX];
LL ans;
int k;
void dfs1(int u,int fa){
    if(vis[u]) num[u]=1;
    for(int i=0;i<v[u].size();i++){
        int vv=v[u][i];
        if(vv!=fa){
            dfs1(vv,u);
            num[u]+=num[vv];
        }
    }
}

void dfs2(int u,int fa){
    for(int i=0;i<v[u].size();i++){
        int vv=v[u][i];
        if(vv!=fa){
            ans+=min(num[vv],2*k-num[vv]);
            dfs2(vv,u);
        }
    }
}
int main(){
    //freopen("in.txt","r",stdin);
    int n;
    while(cin>>n>>k){
        mem(vis,0);mem(num,0);
        for(int i=1;i<=n;i++) v[i].clear();
        for(int i=0;i<2*k;i++){
            int a;
            scanf("%d",&a);
            vis[a]=1;
        }
        for(int i=1;i<n;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            v[a].push_back(b);
            v[b].push_back(a);
        }
        ans=0;
        dfs1(1,-1);
        dfs2(1,-1);
        cout<<ans<<endl;
    }
    return 0;
}