题目链接
题意:给定一棵树,树上有2k个点,现在要给这2k个点两两配对,求这k对点配对后他们的距离之和最大为多少。
思路:又是一个按边算贡献的题,然而自己却总是考虑不到这方面去。。。
我们思考对于一条边u->v他能产生多少贡献?我们要距离和最大的话肯定贪心的向一个点尽可能地往外配对,于是一条边地贡献为min(num【v】,n-num【v】)。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+1;
vector<int>g[maxn];
int n,k,t,u,v,vis[maxn];
ll num[maxn],ans;
void dfs(int u,int fa)
{
num[u]+=vis[u];
for(int to:g[u])
{
if(to==fa) continue;
dfs(to,u);
num[u]+=num[to];
}
ans+=min(num[u],2*k-num[u]);
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=2*k;++i) scanf("%d",&t),vis[t]=1;
for(int i=1;i<n;++i)
{
scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
dfs(1,-1);
printf("%lld\n",ans);
}