【CF700B】B. Connecting Universities （贪心）

连通大学

题意：

• 给定一棵树，和 2k 所大学的位置，大学在树的结点上（不能重叠），要求你给大学两两分组匹配，求 $Max（{\sum }_{i=0}^{2k-1}dis{t}_{i\ast 2,i\ast 2+1}）$
• 即最大的 全部 匹配大学 之间距离总和，距离算的是树上最短路径。

思路：

• 一开始完全没有思路…

• 这个贪心的想法也太极限了。从边的方向思考，考虑一条边能对答案产生的最大贡献.

• 设一条边左边累计的大学数量为 A，右边累计的为 B。

• 贪心地想，这些大学要两两匹配，当两个大学分别在边两端时，这条边就会被走一次。而这条边最大能被走的次数显然就是 $Min(A,B)$。这也就是它的贡献。

• 所以只需要简单 树形dp 预处理下

$Code:$

#include<bits/stdc++.h>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define forr(a,b,c) for(int a=b;a<=c;a++)
#define rfor(a,b,c) for(int a=b;a>=c;a--)
#define all(a) a.begin(),a.end()
#define oper(a) (operator<(const a& ee)const)
#define endl "\n"
using namespace std;

typedef long long ll;
typedef pair<int, int> PII;

double DNF = 1e17;
const int N = 200010, M = 400010, MM = 110;
int INF = 0x3f3f3f3f, mod = 1e9 + 7;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D, K;
int h[N], e[M], ne[M], idx;
int down[N], u[N];
ll ans;

inline void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void dfs_down(int x, int fa) {
	if (u[x])down[x] = 1;
	for (int i = h[x]; ~i; i = ne[i]) {
		int j = e[i];
		if (j == fa)continue;
		dfs_down(j, x);
		down[x] += down[j];
	}
	//对于 x 点到自己父节点的这条边来说，down[x] 维护了 x 以下所有的大学数量
	//那以上就是 总大学数量 - down[x]
	ans += min(down[x], 2 * k - down[x]);
	//累计贡献
}

void solve() {
	cin >> n >> k;
	mem(h, -1);
	forr(i, 1, 2 * k) {
		int a; cin >> a;
		u[a] = 1;
	}
	forr(i, 1, n - 1) {
		int a, b;
		cin >> a >> b;
		add(a, b), add(b, a);
	}
	dfs_down(1, -1);
	cout << ans << endl;
}

int main() {
	cinios;

	T = 1;
	while (T--)solve();

	return 0;
}
/*
*/