【CF1336A】A. Linova and Kingdom（贪心）

利诺瓦和王国

题意：

• 给定一颗树，在树上挑选 K 个结点作为工业城市。
• 每个工业城市能得到的快乐值为：从自身走到根节点 1 的最短路径上能经过的 非工业城市 的个数。
• 求最大能获得的总快乐值。

思路：

• 经典的 K 城市放置，代表这题很有可能是个贪心。
• 我们很简单能想到，在深度大的点放工业城市，这样更快乐。
• 但也要考虑到 若某点放城市，它会影响它以下的所有点，让它们的快乐值都 -1。
• 综合以上，我们需要的是深度尽可能大的点，同时自身以下结点个数少的点（影响小）。
• 根据这个排序即可。可以发现 dep - son 就是该处放城市所能得到的 贡献

$Code:$

#include<bits/stdc++.h>
#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"
#define ul (u << 1) 
#define ur (u << 1 | 1)
using namespace std;

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

double DNF = 1e17;
const int N = 200010, M = 200010, MM = 110;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int INF = 0x3f3f3f3f, mod = 1e9 + 7;
int n, m, k, T, S, D, K;
int son[N], dep[N], z[N];
vector<int> e[N];

int dfs(int x, int fa, int deep) {
	dep[x] = deep;
	//记录某点的dep，和以下点的个数
	for (auto j : e[x]) {
		if (j == fa)continue;
		son[x] += dfs(j, x, deep + 1) + 1;
	}
	return son[x];
}

bool cmp(int x, int y) {
	return x > y;
}

void solve() {
	cin >> n >> k;
	forr(i, 2, n) {
		int a, b;
		cin >> a >> b;
		e[a].push_back(b);
		e[b].push_back(a);
	}
	dfs(1, 0, 0);

	forr(i, 1, n)z[i] = dep[i] - son[i];//根据贡献从大到小排序
	sort(z + 1, z + 1 + n, cmp);

	ll sum = 0;
	forr(i, 1, k)sum += z[i];

	cout << sum;
}

int main() {
	cinios;
	T = 1;
	while (T--)solve();
	return 0;
}
/*
*/