Codeforces Round #635 (Div. 2) C. Linova and Kingdom

Codeforces Round #635 (Div. 2)的其他题解请点这里

C. Linova and Kingdom

题目大意：

有n个城市，n-1条无向边，保证联通
你要规划好k个城市为工业城，其他城市为旅游城

现在要你求每个工业城到主城1（节点1）所获得的权值之和（每个工业城经过旅游城时权值加一）

问最大权值和

---

思路：

很容易想到，为了使权值和最大，工业城要尽可能的建立在叶子节点上，建立完后假设所有叶子节点删除，再找叶子节点

我们用dep[ i ]表示 i 节点的深度(根节点为1)，son[ i ]表示 i 节点的子节点个数（包含自身）
那么每个节点带来的权值就是dep[i] - 1，但是在该节点建立工业城会导致其子节点的权值减一，所以我们要再减去其子节点个数son[ i ] - 1

所以每个节点的权值就是dep[ i ] - son[ i ]
先dfs一遍，然后根据dep[ i ] - son[ i ]排序，取前k大即可

AC代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<map>
#include<random>
#include<algorithm>
#define ll long long 
using namespace std;
const int maxn = 1e6 + 7;
int n, k;
struct e_node {
	int to;
	int next;
}e[maxn];
int tot, head[maxn];
void add_e(int u, int v) {
	e[++tot].to = v;
	e[tot].next = head[u];
	head[u] = tot;
}
struct a_node {
	int to;
	int id;
}a[maxn];
int dep[maxn], son[maxn];
void dfs(int u, int fa) {
	dep[u] = dep[fa] + 1;
	son[u] = 1;
	for (int i = head[u]; i; i = e[i].next) {
		int v = e[i].to;
		if (v == fa)continue;
		dfs(v, u);
		son[u] += son[v];
		a[u].to++;
	}
}

bool cmp(a_node a, a_node b) {
	return dep[a.id] - son[a.id] > dep[b.id] - son[b.id];
}
int main() {
	scanf("%d %d", &n, &k);
	for (int i = 1; i <= n - 1; ++i) {
		int u, v;
		scanf("%d %d", &u, &v);
		add_e(u, v);
		add_e(v, u);
	}
	for (int i = 1; i <= n; ++i)a[i].id = i;
	ll ans = 0;
	dfs(1, 0);
	sort(a + 1, a + n + 1, cmp);

	for (int i = 1; i <= n && k; ++i) {
		ans += dep[a[i].id];
		ans -= son[a[i].id];
		k--;
	}
	printf("%lld\n", ans);
}