题意
有一颗数,树的边权为1,树上的有些节点有大学,现在要把这些大学配对,每对大学之间都要链接线缆,问最多要链接多长的线缆。
图示
以第一组样例为例:
思路
直接dfs去计数每个节点的子节点包含有多少个大学,这样就可以算出每条边的两边有多少大学。
在每条边上能建立的最多的线缆数就是:两端点的子树中包含的大学数量中,较小的那一个。即是说包含较少的大学的端点的子树中的大学都和另外一边的大学配对。对每条边都这样统计一下,求和就是答案
代码
#include <bits/stdc++.h>
using namespace std;
const int maxm = 2e5 + 10;
const int maxn = 2e5 + 10;
typedef long long ll;
struct Edges{
int u,v;
}edge[maxm];
vector<int> G[maxn];
int cnt[maxn];
int dfs(int cur,int from){
for(int i = 0 ; i < G[cur].size() ; i ++){
int v = G[cur][i];
if(v == from) continue;
cnt[cur] += dfs(v,cur);
}
return cnt[cur];
}
int main (){
int n,k;
scanf("%d %d", &n,&k);
for(int i = 1 ; i <= k*2 ; i ++){
int tmp;
scanf("%d", &tmp);
cnt[tmp] = 1;
}
int cn = 0;
for(int i = 0 ; i < n-1 ; i ++){
int u,v;
scanf("%d %d", &u,&v);
edge[i].u = u;
edge[i].v = v;
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,-1);
ll exp = 0;
for(int i = 0 ; i < n-1 ; i ++){
Edges & e = edge[i];
int tt = min(cnt[e.u],cnt[e.v]);
int tmp = min(2*k-tt,tt);
exp += tmp;
}
printf("%I64d\n",exp);
}