CF-Round #635-div2-C题

CF-Round #635-div2-C题

C. Linova and Kingdom

传送门

这道题是dfs思维题。
然后解锁了stl一个函数nth_element();

题目大意：给你一棵树，需要选择k个结点发展工业，其余的n-k个结点是发展旅游业的。首都在1号结点，每次召开会议发展工业的城市需要派一个使者去到首都，在使者当前城市到达首都的简单路径中，遇到旅游城市的个数就是这个使者增加的开心值，问怎样选择发展工业的城市可以使得总的开心值最大。

题目思路：当初我就是想的是只要选择尽可能深的叶子结点发展工业就可以获得最大的开心值。后来往深想了想。还是转化到旅游城市的思路上比较好考虑。

我们考虑当前操作结点u：如果把u用来发展旅游城市代替发展工业城市，那我们的幸福值会增加多少？
这里说明一下：如果我们选择一个结点发展旅游城市，那么它的父结点也是发展旅游城市，这样可以让我们获得最大的幸福值。（你想想，如果你让一个发展旅游城市的父结点发展工业城市，那那个旅游城市在下面有什么意义呢？）
size[u]代表以u为根的子树结点的个数（算结点的时候包括u自己）
dep[u]代表u到根结点的深度（根结点的深度为1开始计数）

当结点u用来发展旅游城市的时候（假设原先是发展工业的，那么它的祖祖先先都是发展旅游的），我们相应的会失去dep[u] - 1个幸福值，相应的，我们会增加size[u] - 1个幸福值（因为当前结点发展旅游城市后，以u为根的子树需要经过u，会增加幸福值）
所以我们幸福感的净增加值是size[u] - 1 - (dep[u] - 1) = size[u] - dep[u];

所以我们只需要dfs跑一遍获得size[], dep[], happ[]就行了~

我们最后把happ[]排序以下，获得最大的n-k个值，维护ans即可。

这里解锁了一个stl：nth_element(）；
用法：nth_element(排序对象的开头，需要排好序的中间位置，排序对象的结尾）；
这里会把前两个参数的对应位置排好序，然后后面就不管了。
（说实话跟sort差不多把，一个是部分排序，一个是整体排序）
可以降低时间复杂度！

代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;

int n;
int k;
vector<int> v[N];
int dep[N]; 
int cnt[N];
int favor[N];
ll ans;

bool cmp(int a, int b)
{
	return a > b;
}

int dfs(int now, int pre)
{
	dep[now] = dep[pre] + 1;
	cnt[now] = 1;
	int siz = v[now].size();
	for (int i = 0; i < siz; i++)
	{
		if (v[now][i] == pre)
		{
			continue;
		}
		cnt[now] += dfs(v[now][i], now);
	}
	favor[now] = cnt[now] - dep[now];
	return cnt[now];
}

int main()
{
	scanf ("%d%d", &n, &k);
	for (int i = 1; i < n; i++)
	{
		int u1, u2;
		scanf ("%d%d", &u1, &u2);
		v[u1].push_back(u2);
		v[u2].push_back(u1);
	}
	dfs(1, 0);
	nth_element(favor + 1, favor + n - k, favor + n + 1, cmp);
	for (int i = 1; i <= n - k; i++)
	{
		ans += favor[i];
	}
	cout << ans << endl;
	return 0;
}